| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_split_op.cc | 51 const int64 input_rank = input_shape.vec<int64>().size(); in Compute() local
|
| D | sparse_concat_op.cc | 77 const int input_rank = input_shape.dims(); in Compute() local
|
| D | reshape_util.cc | 44 const int64 input_rank = input_shape.dims(); in operator ()() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 125 const int input_rank = input_shape.dims(); in SetMatrixDiag() local
358 const int input_rank = input_shape.dims(); in Compile() local
468 const int input_rank = input_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 | data_format_ops.cc | 109 int input_rank = input_tensor_shape.dims(); in Compile() 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/common_runtime/ |
| D | eval_const_tensor.cc | 57 int input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
85 int32 input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 90 int32_t input_rank = input_type.getShape().size(); in convertPackOp() local
269 int64_t input_rank = input_shape.size(); in convertUnpackOp() local
577 auto input_rank = input_type.getShape().size(); in convertConcatV2Op() local
828 int input_rank = input_type.getRank(); in convertSpaceToBatchNDOp() local
1078 int input_rank = input_type.getRank(); in convertBatchToSpaceNDOp() local
1370 int32_t input_rank = output_type.getShape().size(); in convertEluOp() local
1423 int32_t input_rank = input_type.getShape().size(); in convertSoftmaxOp() local
1644 int32_t input_rank = input_type.getShape().size(); in convertLogSoftmaxOp() local
2048 int32_t input_rank = input_type.getRank(); in convertStridedSliceOp() local
2320 auto input_rank = input_shape.size(); in convertReduceOpCommon() local
[all …]
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 234 const int input_rank = input->dims->size; in DefaultLayerNormFloat() local
262 const int input_rank = input->dims->size; in DefaultLayerNorm() local
|
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| 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/ops/ |
| D | array_ops.cc | 1472 int64 input_rank = c->Rank(input); in UniqueIdxShapeFn() local
1629 const int32 input_rank = c->Rank(input); in __anon42d741192402() local
1963 const Tensor* paddings_t, int64 input_rank) { in MirrorPadKnown()
1999 int64 input_rank = c->Value(pad_0); in __anon42d741192c02() local
2137 const int32 input_rank = c->Rank(input); in __anon42d741193002() local
|
| D | math_ops.cc | 1054 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 __anonea01102b1702() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1290 const int32 input_rank = c->Rank(input_shape); in MatrixDiagPartV2Shape() local
1348 const int32 input_rank = c->Rank(input_shape); in MatrixDiagV2Shape() local
1444 int32 input_rank = c->Rank(input_shape); in MatrixSetDiagV2Shape() local
1807 const int32 input_rank, in ReductionShapeHelper()
1857 const int32 input_rank = c->Rank(input); in ReductionShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 838 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
1040 const int input_rank = input_ty.getRank(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 991 int64_t input_rank = input_ty.getRank(); in CanBeTranslatedToDynamicSlice() local
1021 int64_t input_rank = input_ty.getRank(); in TFSliceSizes2HLOSliceSizes() local
2921 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3037 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3107 int64_t input_rank = input_ty.getRank(); in rewriteWithConstantBegin() local
3236 const int input_rank = input_shape.size(); in rewriteWithUnknownBegin() local
4744 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
4994 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | array_grad.cc | 472 auto input_rank = Rank(scope, input); in SliceGrad() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix.cc | 580 std::vector<int64>* numeric_config) -> StatusOr<int64> { in ParseEinsumString()
|