| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | permute_op.cc | 49 const TensorShape input_tensor_shape = ctx->InputShape(0); in Compile() local
50 int input_rank = input_tensor_shape.dims(); in Compile()
54 input_tensor_shape.DebugString())); in Compile()
56 ctx, input_tensor_shape.dim_size(0) == 4, in Compile()
59 input_tensor_shape.DebugString())); in Compile()
62 ctx, input_tensor_shape.dim_size(1) == 2, in Compile()
65 input_tensor_shape.DebugString())); in Compile()
|
| D | conv_ops.cc | 101 TensorShape input_tensor_shape; in Compile() local
102 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsShape(0, &input_tensor_shape)); in Compile()
104 TensorShapeToXLAShape(ctx->input_xla_type(1), input_tensor_shape); in Compile()
|
| D | batchtospace_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in BatchToSpace() argument
28 const int input_rank = input_tensor_shape.dims(); in BatchToSpace()
30 input_tensor_shape.dim_sizes(); in BatchToSpace()
|
| D | spacetobatch_op.cc | 25 DataType input_dtype, const TensorShape& input_tensor_shape, in SpaceToBatch() argument
28 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch()
30 input_tensor_shape.dim_sizes(); in SpaceToBatch()
|
| D | conv_op_helpers.cc | 208 TensorShape input_tensor_shape, filter_tensor_shape, in ConvBackpropComputeDimensionsV2XlaShapes() local
210 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(input_shape, &input_tensor_shape)); in ConvBackpropComputeDimensionsV2XlaShapes()
215 label, num_spatial_dims, input_tensor_shape, filter_tensor_shape, in ConvBackpropComputeDimensionsV2XlaShapes()
410 TensorShape input_tensor_shape, filter_tensor_shape, output_tensor_shape; in MakeXlaBackpropFilterConvOp() local
412 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(input_shape, &input_tensor_shape)); in MakeXlaBackpropFilterConvOp()
|
| /external/tensorflow/tensorflow/contrib/kernel_methods/python/mappers/ |
| D | random_fourier_features.py | 122 input_tensor_shape = input_tensor.get_shape()
123 if len(input_tensor_shape) != 2:
126 len(input_tensor_shape))
128 features_dim = input_tensor_shape[1]
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/ops/ |
| D | array_ops.cc | 32 shape_inference::ShapeHandle input_tensor_shape = c->input(0); in __anon96b295990102() local
34 c->NumElements(input_tensor_shape); in __anon96b295990102()
40 const int rank = c->Rank(input_tensor_shape); in __anon96b295990102()
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/kernels/ |
| D | periodic_resample_op.h | 208 const tensorflow::TensorShape& input_tensor_shape, in process_desired_shape() argument
215 const int rank = input_tensor_shape.dims(); in process_desired_shape()
227 context, desired_shape[i] >= input_tensor_shape.dim_size(i), in process_desired_shape()
232 i, " input tensor has size ", input_tensor_shape.dim_size(i), in process_desired_shape()
245 input_tensor_shape.num_elements() / new_sliced_size; in process_desired_shape()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_pooling_ops_common.h | 511 const TensorShape& input_tensor_shape) { in InitMklPoolParameters() argument
514 this->data_format_tf_, input_tensor_shape); in InitMklPoolParameters()
611 TensorShape input_tensor_shape = input_tensor.shape(); in ConfigureInput() local
618 ? TFShapeToMklDnnDimsInNCHW(input_tensor_shape, in ConfigureInput()
620 : TFShapeToMklDnnDimsInNCDHW(input_tensor_shape, in ConfigureInput()
637 input_tensor_shape); in ConfigureInput()
|
| D | mkl_maxpooling_op.cc | 529 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
531 input_tensor_shape); in Compute()
553 input_tensor_shape, this->data_format_tf_), in Compute()
557 input_tensor_shape, this->data_format_tf_), in Compute()
|
| D | mkl_avgpooling_op.cc | 459 TensorShape input_tensor_shape = input_tensor.shape(); in Compute() local
461 input_tensor_shape); in Compute()
|
| /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
1363 HloInstruction::CreateParameter(0, input_tensor_shape, "input_tensor")); in TEST_F()
1390 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1398 HloInstruction::CreateParameter(0, input_tensor_shape, "input_tensor")); in TEST_F()
1425 Shape input_tensor_shape = ShapeUtil::MakeShape(F32, {50, 49, 48, 47, 46}); in TEST_F() local
1433 HloInstruction::CreateParameter(0, input_tensor_shape, "input_tensor")); in TEST_F()
1453 input_tensor_shape, input, scatter_indices, scatter_updates, in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 159 TensorShapeProto* input_tensor_shape,
|
| D | constant_folding.cc | 2465 TensorShapeProto* input_tensor_shape, TensorShapeProto* output_tensor_shape, in IsReductionCandidateForSimplification() argument
2489 *input_tensor_shape = input_props.shape(); in IsReductionCandidateForSimplification()
2491 for (int i = 0; i < input_tensor_shape->dim_size(); ++i) { in IsReductionCandidateForSimplification()
2492 if (input_tensor_shape->dim(i).size() < 0) { in IsReductionCandidateForSimplification()
2502 TensorShape(*input_tensor_shape).num_elements(); in IsReductionCandidateForSimplification()
2545 TensorShapeProto input_tensor_shape, output_tensor_shape; in SimplifyReduction() local
2547 *node, properties, &input_tensor_shape, &output_tensor_shape, in SimplifyReduction()
2573 *node, input_tensor_shape, keep_dims, reduction_indices_vector); in SimplifyReduction()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 761 input_tensor_shape = tf_utils.convert_shapes(input_shape, to_tuples=False)
762 batch_size = nest.flatten(input_tensor_shape)[0][0] if input_shape else None
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 447 // for (k : [0, input_tensor_shape.rank))
|