Searched refs:ShuffleInputDataTypeAttribute (Results 1 – 1 of 1) sorted by relevance
40 std::vector<DataType> ShuffleInputDataTypeAttribute( in ShuffleInputDataTypeAttribute() function297 ShuffleInputDataTypeAttribute(types, index_mapping); in MaybeRewriteWhileNode()426 ShuffleInputDataTypeAttribute(in_types, index_mapping); in MaybeRewriteIfNode()