| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | slice_spmd_expander.cc | 181 int64_t input_rank; in GetInputOrientedData() local
386 const int input_rank, in ApplyNewAndShrinkMasksToLayout()
469 const int64_t input_rank = input_type.getRank(); in ExpandOp() local
615 const int input_rank = global_input_shape.size(); in ExpandOp() local
655 const int input_rank = global_input_shape.size(); in ComputeLayoutForward() local
681 const int input_rank = global_input_shape.size(); in ComputeLayoutBackward() local
726 const int input_rank = global_input_shape.size(); in ExpandOp() local
790 const int input_rank = global_input_shape.size(); in ComputeLayoutForward() local
826 const int input_rank = global_input_shape.size(); in ComputeLayoutBackward() local
872 const int input_rank = ValueRank(strided_slice_grad_op.dy()); in ExpandOp() local
[all …]
|
| D | argmax_spmd_expander.cc | 47 const auto input_rank = ValueRank(argmax_op.input()); in ComputeResultLayout() local
77 const auto input_rank = ValueRank(input); in ExpandOp() local
147 const auto input_rank = ValueRank(input); in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util_gpu.cu.cc | 31 const Tindex nnz, const Tindex input_rank, const Tindex output_rank, in ReshapeSparseTensorKernel()
63 const int64 input_rank = input_shape.dims(); in operator ()() local
|
| D | sparse_concat_op.cc | 48 const int input_rank = input_shape.dims(); in operator ()() local
154 const int input_rank = input_shape.dims(); in Compute() local
|
| D | reshape_util.cc | 48 const int64_t input_rank = input_shape.dims(); in operator ()() local
|
| D | sparse_split_op.cc | 104 const int64_t input_rank = input_shape.vec<int64_t>().size(); in SparseSplitOpImpl() local
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | grad_helper.cc | 48 auto input_rank = Size(scope, input_shape); in ReducedShapeHelper() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | matrix_diag_ops.cc | 128 const int input_rank = input_shape.dims(); in SetMatrixDiag() local
361 const int input_rank = input_shape.dims(); in Compile() local
471 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 | 29 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch() local
|
| D | quantize_and_dequantize_op.cc | 81 int64_t input_rank = input_shape.dims(); in Compile() local
|
| /external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| D | layer_norm.cc | 287 const int input_rank = input->dims->size; in IntegerLayerNorm() local
330 const int input_rank = input->dims->size; in DefaultLayerNormFloat() local
358 const int input_rank = input->dims->size; in DefaultLayerNorm() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
| D | data_format_vec_permute.cc | 69 int input_rank = x_dims.nbDims; in Validate() 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 | 93 const int input_rank = input_shape.dims(); in Compute() local
|
| D | linalg_ops_common.cc | 123 int input_rank = -1; in AnalyzeInputs() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 318 int64_t input_rank = input_shape.size(); in convertUnpackOp() local
766 int input_rank = input_type.getRank(); in convertSpaceToBatchNDOp() local
1018 int input_rank = input_type.getRank(); in convertBatchToSpaceNDOp() local
1326 int32_t input_rank = output_type.getShape().size(); in convertEluOp() local
1380 int32_t input_rank = input_type.getShape().size(); in convertSoftmaxOp() local
1811 int32_t input_rank = input_type.getShape().size(); in convertLogSoftmaxOp() local
2289 int32_t input_rank = input_type.getRank(); in convertStridedSliceOp() local
2611 int64_t input_rank = input_shape.size(); in convertGenericReduceOp() local
2663 auto input_rank = input_shape.size(); in convertReduceOpCommon() local
2899 int64_t input_rank = input_type.getRank(); in convertReduceMeanOp() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | roll.cc | 205 const int input_rank = NumDimensions(input); in Eval() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1495 int64_t input_rank = c->Rank(input); in UniqueIdxShapeFn() local
1652 const int32_t input_rank = c->Rank(input); in __anon091a71b72402() local
1996 const Tensor* paddings_t, int64_t input_rank) { in MirrorPadKnown()
2032 int64_t input_rank = c->Value(pad_0); in __anon091a71b72c02() local
2170 const int32_t input_rank = c->Rank(input); in __anon091a71b73002() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 1381 const int32_t input_rank = c->Rank(input_shape); in MatrixDiagPartV2Shape() local
1439 const int32_t input_rank = c->Rank(input_shape); in MatrixDiagV2Shape() local
1535 int32_t input_rank = c->Rank(input_shape); in MatrixSetDiagV2Shape() local
1898 const int32_t input_rank, in ReductionShapeHelper()
1948 const int32_t input_rank = c->Rank(input); in ReductionShape() local
|
| /external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| D | onedevice_train_palm_detection_filter_inference_beta.cc | 893 void StridedSlice(const int32_t input_rank, in StridedSlice()
1043 void OneHot(const int32_t input_rank, in OneHot()
1070 void OneHotLastDim(const int32_t input_rank, in OneHotLastDim()
1271 inline int32_t BroadcastOffset(int32_t input_rank, in BroadcastOffset()
|