| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | permute_op.cc | 50 int input_rank = input_tensor_shape.dims(); in Compile() local
|
| D | depthtospace_op.cc | 62 int input_rank = input_shape.size(); in Compile() local
|
| D | spacetodepth_op.cc | 62 int input_rank = input_shape.size(); in Compile() local
|
| D | batchtospace_op.cc | 28 const int input_rank = input_tensor_shape.dims(); in BatchToSpace() local
|
| D | spacetobatch_op.cc | 28 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 53 int input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
81 int32 input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_concat_op.cc | 77 const int input_rank = input_shape.dims(); in Compute() local
|
| D | reshape_util.cc | 50 const int64 input_rank = input_shape_in.NumElements(); in Reshape() local
|
| D | lu_op.cc | 71 int input_rank = input.dims(); in Compute() local
|
| D | lu_op_gpu.cu.cc | 85 const int input_rank = input.dims(); in ComputeAsync() local
|
| D | linalg_ops_common.cc | 118 int input_rank = -1; in AnalyzeInputs() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1490 const int32 input_rank = c->Rank(input); in __anon7c94107b2402() local
1808 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown()
1844 int64 input_rank = c->Value(pad_0); in __anon7c94107b2c02() local
1982 const int32 input_rank = c->Rank(input); in __anon7c94107b3002() local
|
| D | math_ops.cc | 917 const int32 input_rank = c->Rank(input_shape); in ArgOpShape() local
|
| D | data_flow_ops.cc | 638 int32 input_rank = c->Rank(input_shape); in __anon4a7ac08d1302() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 463 auto input_rank = Rank(scope, input); in SliceGrad() local
|
| D | math_grad.cc | 644 auto input_rank = Size(scope, input_shape); in ReducedShapeHelper() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1133 const int32 input_rank, in ReductionShapeHelper()
1183 const int32 input_rank = c->Rank(input); in ReductionShape() local
|
| D | shape_inference.cc | 984 int idx, int input_rank, DimensionHandle* out) { in MakeDimForScalarInputWithNegativeIndexing()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1302 int64 input_rank = input_shape.rank(); in AlignLayouts() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2190 const int input_rank = input_dims.size(); in ConvertExpandDims() local
2242 const int input_rank = input_dims.size(); in ConvertSqueeze() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 572 int input_rank = input_shape.dimensions_count(); in ProcessTensorFlowReductionOperator() local
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 831 int16 input_rank; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 691 const int input_rank = input_prop.shape().dim_size(); in MaterializeReductionIndices() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 802 int input_rank, const int32* coords_data, in Gather()
|