Searched defs:indices_type (Results 1 – 14 of 14) sorted by relevance
152 #define REGISTER_CPU_KERNEL(indices_type, num_segments_type) \ argument
273 DataType indices_type = context->input_type(1); in Compile() local
224 const DataType indices_type = input_type(1); in Compile() local
398 DataType indices_type = ctx->input_type(1); in Compile() local
55 auto indices_type = RankedTensorType::get(input_shape, builder->getI64Type()); in createMaxUnpoolingFunc() local
37 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel()
972 ShapedType &indices_type, in NormalizeIndexVector()1098 ShapedType indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local
796 auto indices_type = scatter.indices().getType().cast<RankedTensorType>(); in HandleTensorListScatterIntoExistingListOp() local
182 auto indices_type = indices.getType().dyn_cast<RankedTensorType>(); in CanOptimizeIdentityGatherNdOrScatterNdOp() local
969 auto indices_type = indices.getType().cast<TensorType>(); in BuildGatherOp() local1035 auto indices_type = indices.getType().cast<ShapedType>(); in Verify() local
238 auto indices_type = value.getType().cast<RankedTensorType>(); in UnpackTensorAlongZeroDim() local5068 auto indices_type = in matchAndRewrite() local5180 auto indices_type = indices.getType().cast<ShapedType>(); in matchAndRewrite() local5240 auto indices_type = op.indices().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
1902 tflite::SparseIndexVector indices_type; in BuildSparsityParameters() local
1825 const tensorflow::DataType indices_type = in ConvertReduceOperator() local
2061 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather() local