Searched refs:shuffled_shape (Results 1 – 3 of 3) sorted by relevance
116 TensorShape shuffled_shape();226 helper.shuffled_shape(),
35 TensorShape ReductionHelper::shuffled_shape() { in shuffled_shape() function in tensorflow::ReductionHelper
132 Shape shuffled_shape; in ExportFloatArray() local134 &shuffled_shape); in ExportFloatArray()136 ShuffleArray(input_shape, input_axes_order, output_axes_order, shuffled_shape, in ExportFloatArray()139 ExportFloatArray(shuffled_shape, shuffled_data.data(), output_tensor, in ExportFloatArray()