| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | eval_const_tensor.cc | 56 int input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
84 int32 input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 125 const int input_rank = input_shape.dims(); in SetMatrixDiag() local
357 const int input_rank = input_shape.dims(); in Compile() local
467 const int input_rank = input_shape.dims(); in Compile() local
|
| D | data_format_ops.cc | 104 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
|
| D | quantize_and_dequantize_op.cc | 81 int64 input_rank = input_shape.dims(); in Compile() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 50 const int64 input_rank = input_shape_in.NumElements(); in Reshape() local
|
| D | sparse_concat_op.cc | 77 const int input_rank = input_shape.dims(); in Compute() local
|
| D | lu_op.cc | 71 int input_rank = input.dims(); in Compute() local
|
| D | lu_op_gpu.cu.cc | 87 const int input_rank = input.dims(); in ComputeAsync() local
|
| D | matrix_set_diag_op.cc | 90 const int input_rank = input_shape.dims(); in Compute() local
|
| D | linalg_ops_common.cc | 120 int input_rank = -1; in AnalyzeInputs() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1237 const int32 input_rank = c->Rank(input_shape); in MatrixDiagPartV2Shape() local
1295 const int32 input_rank = c->Rank(input_shape); in MatrixDiagV2Shape() local
1391 int32 input_rank = c->Rank(input_shape); in MatrixSetDiagV2Shape() local
1704 const int32 input_rank, in ReductionShapeHelper()
1754 const int32 input_rank = c->Rank(input); in ReductionShape() local
|
| D | shape_inference.cc | 924 int idx, int input_rank, DimensionHandle* out) { in MakeDimForScalarInputWithNegativeIndexing()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 577 int64_t input_rank = input_ty.getRank(); in CanBeTranslatedToDynamicSlice() local
624 int64_t input_rank = input_ty.getRank(); in TFSliceSizes2HLOSliceSizes() local
1588 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
1704 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
1785 int64_t input_rank = input_ty.getRank(); in matchAndRewrite() local
2975 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3218 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1579 const int32 input_rank = c->Rank(input); in __anonf6523ebd2302() local
1913 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown()
1949 int64 input_rank = c->Value(pad_0); in __anonf6523ebd2b02() local
2087 const int32 input_rank = c->Rank(input); in __anonf6523ebd2f02() local
|
| D | math_ops.cc | 1037 const int32 input_rank = c->Rank(input_shape); in ArgOpShape() local
|
| D | data_flow_ops.cc | 693 int32 input_rank = c->Rank(input_shape); in __anonb84a3fcf1702() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 463 auto input_rank = Rank(scope, input); in SliceGrad() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 546 std::vector<int64>* numeric_config) -> StatusOr<int64> { in ParseEinsumString()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 635 auto input_rank = rewriter->create<TF::RankOp>( in CreateCondFalseBranch() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops.cc | 1012 int64_t input_rank = input_ty.getRank(); in InferExpandDimsOpType() local
2057 int64_t input_rank = input_type.getRank(); in VerifySplitInputAndSplitDim() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 1329 int64 input_rank = input_shape.rank(); in AlignLayouts() local
|