| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | materialize_broadcasts.cc | 47 ArrayRef<int64_t> operandShape = operandType.getShape(); in matchAndRewrite() local
|
| D | hlo_legalize_to_memref.cc | 230 auto operandShape = operandType.getShape(); in insertDynamicMemrefCastOp() local
|
| D | unfuse_batch_norm.cc | 207 Value operandShape = getShapeValue(op->getLoc(), operand, rewriter); in calculateReduceSize() local
|
| D | legalize_to_linalg.cc | 695 auto operandShape = operandType.getShape(); in getIndexingMaps() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/gml_st/transforms/ |
| D | fusion_interface_impl.cc | 55 const auto& operandShape = in operandMaterializesToPoint() local
72 auto operandShape = operand.getType().cast<RankedTensorType>().getShape(); in buildPointOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/ |
| D | shape_simplification.cc | 120 llvm::count_if(operandExtents, [&](ArrayRef<int64_t> operandShape) { in matchAndRewrite()
|
| D | symbolic_shape_optimization.cc | 621 ArrayRef<int64_t> operandShape, ArrayRef<SymbolicExpr> operandShapeInfo) { in concretizeOperandShape()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/ |
| D | lhlo_legalize_to_affine.cc | 138 ArrayRef<int64_t> operandShape = operandType.getShape(); in matchAndRewrite() local
252 ArrayRef<int64_t> operandShape = operandType.getShape(); in matchAndRewrite() local
516 ArrayRef<int64_t> operandShape = operandType.getShape(); in matchAndRewrite() local
|
| D | lhlo_legalize_to_parallel_loops.cc | 112 auto operandShape = operand.getType().template cast<MemRefType>().getShape(); in mapWindowIvsToInput() local
244 auto operandShape = operand.getType().cast<MemRefType>().getShape(); in createReduceOpInNestedParallelLoops() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.cpp | 1242 auto operandShape = operandType.getShape(); in inferReturnTypeComponents() local
1345 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyGather()
1453 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyStaticGather()
1554 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in inferGatherReturnTypeComponents()
1665 ShapeAdaptor operandShape = operands.getShape(0); in inferReturnTypeComponents() local
1711 ShapeAdaptor operandShape = operands.getShape(0); in inferReturnTypeComponents() local
2499 auto operandShape = operandType.getShape(); in verify() local
2813 auto operandShape = operandType.getShape(); in verify() local
5442 auto operandShape = operandTypes[i].getShape(); in verify() local
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 1279 auto operandShape = operandType.getShape(); in inferReturnTypeComponents() local
1457 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyGather()
1565 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in verifyStaticGather()
1666 ShapeAdaptor operandShape, ShapeAdaptor startIndicesShape, in inferGatherReturnTypeComponents()
1777 ShapeAdaptor operandShape = operands.getShape(0); in inferReturnTypeComponents() local
1823 ShapeAdaptor operandShape = operands.getShape(0); in inferReturnTypeComponents() local
2097 auto operandShape = this->operand().getType().cast<RankedTensorType>(); in fold() local
3016 auto operandShape = operandType.getShape(); in verify() local
3526 auto operandShape = operandType.getShape(); in verify() local
8024 auto operandShape = operandTypes[i].getShape(); in verify() local
|
| /external/android-nn-driver/ |
| D | ConversionUtils.hpp | 2049 armnn::TensorShape operandShape = GetTensorShapeForOperand(*operand); in ConvertConcatenation() local
|
| /external/armnn/shim/sl/canonical/ |
| D | Converter.cpp | 730 armnn::TensorShape operandShape = GetTensorShapeForOperand(*operand); in ConvertConcatenation() local
|