| /external/tensorflow/tensorflow/core/kernels/ |
| D | unsorted_segment_join_op.cc | 154 #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 | 273 DataType indices_type = context->input_type(1); in Compile() local
|
| D | tensor_list_ops.cc | 400 DataType indices_type = ctx->input_type(1); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | perception_ops_utils_test.cc | 54 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 | 1577 ShapedType &indices_type, in NormalizeIndexVector()
1773 ShapedType indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local
|
| D | tensor_list_ops_decomposition.cc | 780 auto indices_type = scatter.indices().getType().cast<RankedTensorType>(); in HandleTensorListScatterIntoExistingListOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_common.cc | 3124 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertGatherOp() local
3380 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertGatherNdOp() local
3548 auto indices_type = indices_value.getType().dyn_cast<RankedTensorType>(); in convertOneHotOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 212 auto indices_type = indices.getType().dyn_cast<RankedTensorType>(); in CanOptimizeIdentityGatherNdOrScatterNdOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 1186 auto indices_type = indices.getType().cast<TensorType>(); in BuildGatherOp() local
1274 auto indices_type = indices.getType().cast<ShapedType>(); in Verify() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 215 auto indices_type = value.getType().cast<RankedTensorType>(); in UnpackTensorAlongZeroDim() local
6183 auto indices_type = in matchAndRewrite() local
6299 auto indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local
6359 auto indices_type = op.indices().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_export.cc | 2029 tflite::SparseIndexVector indices_type; in BuildSparsityParameters() local
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1825 const tensorflow::DataType indices_type = in ConvertReduceOperator() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 2123 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather() local
|