Searched defs:split_dtype (Results 1 – 3 of 3) sorted by relevance
34 int output_ragged_rank, DataType value_dtype, DataType split_dtype, in RaggedComponentsFromVariant()311 const auto split_dtype = DataTypeToEnum<SPLIT_TYPE>::v(); in Compute() local
46 const auto split_dtype = DataTypeToEnum<SPLIT_TYPE>::v(); in BuildDecodeRaggedTensorGraph() local
1008 auto split_dtype = tensor_indices_or_sections.scalar_type(); in tensor_split() local