| /external/tensorflow/tensorflow/core/kernels/ |
| D | unsorted_segment_join_op.cc | 161 #define REGISTER_CPU_KERNEL(indices_type, num_segments_type) \ argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 153 const DataType indices_type = input_type(1); in Compile() local
|
| D | gather_op.cc | 277 DataType indices_type = context->input_type(1); in Compile() local
|
| D | tensor_list_ops.cc | 406 DataType indices_type = ctx->input_type(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 56 auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); in createMaxUnpoolingFunc() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot_test.cc | 37 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 2800 const Value &indices, const ShapedType &indices_type, Value &updates, in CanonicalizeScatterUpdates()
2843 ShapedType &indices_type, in NormalizeIndexVector()
3140 ShapedType indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| D | tensor_list_ops_decomposition.cc | 786 auto indices_type = scatter.indices().getType().cast<RankedTensorType>(); in HandleTensorListScatterIntoExistingListOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 3401 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertGatherOp() local
3670 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertGatherNdOp() local
3837 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertOneHotOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 198 auto indices_type = indices.getType().dyn_cast<RankedTensorType>(); in CanOptimizeIdentityGatherNdOrScatterNdOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 176 auto indices_type = value.getType().cast<RankedTensorType>(); in UnpackTensorAlongZeroDim() local
5903 auto indices_type = in matchAndRewrite() local
6013 auto indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local
6071 auto indices_type = op.indices().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 413 DataType indices_type; member
434 DataType indices_type; member
444 DataType indices_type; member
2920 auto indices_type = DT_INT32; in TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1328 auto indices_type = indices.getType().cast<TensorType>(); in BuildGatherOp() local
1417 auto indices_type = indices.getType().cast<ShapedType>(); in verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 2191 tflite::SparseIndexVector indices_type; in BuildSparsityParameters() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1830 const tensorflow::DataType indices_type = in ConvertReduceOperator() local
|
| /external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| D | pass.cc | 1976 ShapedType indices_type = *(reduction_indices->result_type_begin()); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2368 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather() local
|