Searched defs:input_tensor_shape (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops.cc | 32 shape_inference::ShapeHandle input_tensor_shape = c->input(0); in __anon96b295990102() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | permute_op.cc | 49 const TensorShape input_tensor_shape = ctx->InputShape(0); in Compile() local
|
| D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace()
|
| D | spacetobatch_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch()
|
| D | conv_op_helpers.cc | 208 TensorShape input_tensor_shape, filter_tensor_shape, in ConvBackpropComputeDimensionsV2XlaShapes() local
410 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
|
| D | conv_ops.cc | 101 TensorShape input_tensor_shape; in Compile() local
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 208 const tensorflow::TensorShape& input_tensor_shape, in process_desired_shape()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.h | 511 const TensorShape& input_tensor_shape) { in InitMklPoolParameters()
611 TensorShape input_tensor_shape = input_tensor.shape(); in ConfigureInput() local
|
| D | mkl_avgpooling_op.cc | 459 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
|
| D | mkl_maxpooling_op.cc | 529 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1355 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1390 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1425 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2465 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification()
2545 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local
|