Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
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
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
Dsparse_split_op.cc51 const int64_t input_rank = input_shape.vec<int64>().size(); in Compute() local
Dreshape_util.cc46 const int64_t input_rank = input_shape.dims(); in operator ()() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc127 const int input_rank = input_shape.dims(); in SetMatrixDiag() local
360 const int input_rank = input_shape.dims(); in Compile() local
470 const int input_rank = input_shape.dims(); in Compile() local
Ddepthtospace_op.cc62 int input_rank = input_shape.size(); in Compile() local
Dspacetodepth_op.cc62 int input_rank = input_shape.size(); in Compile() local
Ddata_format_ops.cc109 int input_rank = input_tensor_shape.dims(); in Compile() local
Dbatchtospace_op.cc28 const int input_rank = input_tensor_shape.dims(); in BatchToSpace() local
Dspacetobatch_op.cc28 const int input_rank = input_tensor_shape.dims(); in SpaceToBatch() local
Dquantize_and_dequantize_op.cc81 int64_t input_rank = input_shape.dims(); in Compile() local
/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc57 int input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
85 int32_t input_rank = c->Rank(c->input(0)); in TryToInferTensorOutputFromInputShapes() local
/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/
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
/external/tensorflow/tensorflow/core/kernels/linalg/
Dlu_op_gpu.cu.cc87 const int input_rank = input.dims(); in ComputeAsync() local
Dlu_op.cc71 int input_rank = input.dims(); in Compute() local
Dmatrix_set_diag_op.cc93 const int input_rank = input_shape.dims(); in Compute() local
Dlinalg_ops_common.cc120 int input_rank = -1; in AnalyzeInputs() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc285 int64_t input_rank = input_shape.size(); in convertUnpackOp() local
736 int input_rank = input_type.getRank(); in convertSpaceToBatchNDOp() local
985 int input_rank = input_type.getRank(); in convertBatchToSpaceNDOp() local
1276 int32_t input_rank = output_type.getShape().size(); in convertEluOp() local
1330 int32_t input_rank = input_type.getShape().size(); in convertSoftmaxOp() local
1715 int32_t input_rank = input_type.getShape().size(); in convertLogSoftmaxOp() local
2136 int32_t input_rank = input_type.getRank(); in convertStridedSliceOp() local
2422 auto input_rank = input_shape.size(); in convertReduceOpCommon() local
2631 int64_t input_rank = input_type.getRank(); in convertReduceMeanOp() local
/external/tensorflow/tensorflow/lite/kernels/
Droll.cc205 const int input_rank = NumDimensions(input); in Eval() local
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1485 int64_t input_rank = c->Rank(input); in UniqueIdxShapeFn() local
1642 const int32_t input_rank = c->Rank(input); in __anon5ce23fde2402() local
1976 const Tensor* paddings_t, int64_t input_rank) { in MirrorPadKnown()
2012 int64_t input_rank = c->Value(pad_0); in __anon5ce23fde2c02() local
2150 const int32_t input_rank = c->Rank(input); in __anon5ce23fde3002() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1330 const int32_t input_rank = c->Rank(input_shape); in MatrixDiagPartV2Shape() local
1388 const int32_t input_rank = c->Rank(input_shape); in MatrixDiagV2Shape() local
1484 int32_t input_rank = c->Rank(input_shape); in MatrixSetDiagV2Shape() local
1847 const int32_t input_rank, in ReductionShapeHelper()
1897 const int32_t input_rank = c->Rank(input); in ReductionShape() local
/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
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()
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()
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()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc949 int64_t input_rank = input_ty.getRank(); in CanBeTranslatedToDynamicSlice() local
979 int64_t input_rank = input_ty.getRank(); in TFSliceSizes2HLOSliceSizes() local
1610 int input_rank = input_type.getRank(); in matchAndRewrite() local
1848 int input_rank = input_ty.getRank(); in matchAndRewrite() local
3529 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3592 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3721 int64_t input_rank = input_type.getRank(); in matchAndRewrite() local
3791 int64_t input_rank = input_ty.getRank(); in rewriteWithConstantBegin() local
3920 const int input_rank = input_shape.size(); in rewriteWithUnknownBegin() local
4877 int64_t input_rank = input_ty.getRank(); in matchAndRewrite() local
[all …]

12