Searched defs:startIndicesShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 258 ArrayRef<int64_t> startIndicesShape = startIndicesType.getShape(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.cpp | 1345 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyGather()
1453 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyStaticGather()
1554 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in inferGatherReturnTypeComponents()
1666 ShapeAdaptor startIndicesShape = operands.getShape(1); in inferReturnTypeComponents() local
1712 ShapeAdaptor startIndicesShape = operands.getShape(1); in inferReturnTypeComponents() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1457 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyGather()
1565 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyStaticGather()
1666 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in inferGatherReturnTypeComponents()
1778 ShapeAdaptor startIndicesShape = operands.getShape(1); in inferReturnTypeComponents() local
1824 ShapeAdaptor startIndicesShape = operands.getShape(1); in inferReturnTypeComponents() local
|