MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / HandleGather

Method HandleGather

tensorflow/compiler/xla/service/hlo_verifier.cc:1032–1038  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1030} // namespace
1031
1032Status ShapeVerifier::HandleGather(HloInstruction* gather) {
1033 return CheckShape(
1034 gather,
1035 ShapeInference::InferGatherShape(
1036 gather->operand(0)->shape(), gather->operand(1)->shape(),
1037 gather->gather_dimension_numbers(), gather->gather_slice_sizes()));
1038}
1039
1040Status ShapeVerifier::HandleScatter(HloInstruction* scatter) {
1041 return CheckShape(

Callers 1

VisitMethod · 0.45

Calls 3

shapeMethod · 0.45
operandMethod · 0.45
gather_slice_sizesMethod · 0.45

Tested by

no test coverage detected