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
|