| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | conv_op_helpers.cc | 142 TensorShape input_tensor_shape, filter_tensor_shape, in ConvBackpropComputeDimensionsV2XlaShapes() local
407 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
|
| H A D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace()
|
| H A D | data_format_ops.cc | 108 const TensorShape input_tensor_shape = ctx->InputShape(0); in Compile() local
|
| H A D | spacetobatch_op.cc | 26 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch()
|
| H A D | conv_ops.cc | 106 TensorShape input_tensor_shape; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_pooling_ops_common.h | 516 const TensorShape& input_tensor_shape) { in InitMklPoolParameters()
628 TensorShape input_tensor_shape = input_tensor.shape(); in ConfigureInput() local
|
| H A D | mkl_avgpooling_op.cc | 65 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
|
| H A D | mkl_maxpooling_op.cc | 67 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| H A D | conv_spmd_expander.cc | 116 auto input_tensor_shape = input_tensor_type.getShape(); in PadInputOnUnshardedDim() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | convert_tensor.cc | 209 TensorShape input_tensor_shape(input_tensor.tensor_shape()); in ConvertTensorProto() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/ |
| H A D | convert_tensor.cc | 207 TensorShape input_tensor_shape(input_tensor.tensor_shape()); in ConvertTensorProto() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction_test.cc | 1583 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1619 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1655 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
| H A D | collectives.cc | 455 llvm::ArrayRef<int64_t> input_tensor_shape = input_tensor_type.getShape(); in EmitHaloExchange() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | shape_inference.cc | 1619 auto input_tensor_shape = in PrecheckForXlaConvV2Op() local
1749 auto input_tensor_shape = input_tensor.getType().cast<RankedTensorType>(); in InferShapeForXlaConvV2Op() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 2795 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification()
2890 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local
|