Searched refs:shuffled_shape (Results 1 – 3 of 3) sorted by relevance
123 TensorShape shuffled_shape();223 helper.shuffled_shape(), &shuffled,
35 TensorShape ReductionHelper::shuffled_shape() { in shuffled_shape() function in tensorflow::ReductionHelper
129 Shape shuffled_shape; in ExportFloatArray() local131 &shuffled_shape); in ExportFloatArray()133 ShuffleArray(input_shape, input_axes_order, output_axes_order, shuffled_shape, in ExportFloatArray()136 ExportFloatArray(shuffled_shape, shuffled_data.data(), output_tensor, in ExportFloatArray()