Home
last modified time | relevance | path

Searched defs:input_rank (Results 1 – 25 of 50) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dslice_spmd_expander.cc181 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 …]
H A Dargmax_spmd_expander.cc47 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dmatrix_diag_ops.cc128 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
H A Ddepthtospace_op.cc62 int input_rank = input_shape.size(); in Compile() local
H A Dspacetodepth_op.cc62 int input_rank = input_shape.size(); in Compile() local
H A Dbatchtospace_op.cc28 const int input_rank = input_tensor_shape.dims(); in BatchToSpace() local
H A Ddata_format_ops.cc109 int input_rank = input_tensor_shape.dims(); in Compile() local
H A Dspacetobatch_op.cc29 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch() local
H A Dquantize_and_dequantize_op.cc81 int64_t input_rank = input_shape.dims(); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dreshape_util_gpu.cu.cc31 const Tindex nnz, const Tindex input_rank, const Tindex output_rank, in ReshapeSparseTensorKernel()
63 const int64 input_rank = input_shape.dims(); in operator ()() local
H A Dsparse_concat_op.cc48 const int input_rank = input_shape.dims(); in operator ()() local
154 const int input_rank = input_shape.dims(); in Compute() local
H A Dreshape_util.cc48 const int64_t input_rank = input_shape.dims(); in operator ()() local
H A Dsparse_split_op.cc104 const int64_t input_rank = input_shape.vec<int64_t>().size(); in SparseSplitOpImpl() local
/aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
H A Dlayer_norm.cc287 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_common.cc318 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
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc1495 int64_t input_rank = c->Rank(input); in UniqueIdxShapeFn() local
1652 const int32_t input_rank = c->Rank(input); in __anon38bbb0e82402() local
1996 const Tensor* paddings_t, int64_t input_rank) { in MirrorPadKnown()
2032 int64_t input_rank = c->Value(pad_0); in __anon38bbb0e82c02() local
2170 const int32_t input_rank = c->Rank(input); in __anon38bbb0e83002() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc875 int64_t input_rank = input_ty.getRank(); in CanBeTranslatedToDynamicSlice() local
905 int64_t input_rank = input_ty.getRank(); in TFSliceSizes2HLOSliceSizes() local
1435 int input_rank = input_type.getRank(); in matchAndRewrite() local
1668 int input_rank = input_ty.getRank(); in matchAndRewrite() local
3326 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3389 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3529 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3599 int64_t input_rank = input_ty.getRank(); in rewriteWithConstantBegin() local
3719 const int input_rank = input_shape.size(); in rewriteWithUnknownBegin() local
4727 int64_t input_rank = input_ty.getRank(); in matchAndRewrite() local
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc1381 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
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dgrad_helper.cc48 auto input_rank = Size(scope, input_shape); in ReducedShapeHelper() local
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc893 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()
H A Donedevice_train_palm_detection_filter_inference.cc894 void StridedSlice(const int32_t input_rank, in StridedSlice()
1044 void OneHot(const int32_t input_rank, in OneHot()
1071 void OneHotLastDim(const int32_t input_rank, in OneHotLastDim()
1272 inline int32_t BroadcastOffset(int32_t input_rank, in BroadcastOffset()
H A Donedevice_train_palm_detection_filter_inference_v2.cc893 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()
1272 inline int32_t BroadcastOffset(int32_t input_rank, in BroadcastOffset()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dlower_tf.cc869 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
1068 const int input_rank = input_ty.getRank(); in matchAndRewrite() local
1570 int input_rank = input_shape.size(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Ddata_format_vec_permute.cc69 int input_rank = x_dims.nbDims; in Validate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dlu_op.cc71 int input_rank = input.dims(); in Compute() local

12