Searched defs:output_matrix_shapes (Results 1 – 2 of 2) sorted by relevance
100 TensorShapes output_matrix_shapes; in Compute() local171 TensorShapes* output_matrix_shapes) { in PrepareOutputs()235 const TensorShapes& output_matrix_shapes) { in ComputeTensorSlice()
96 TensorShapes output_matrix_shapes({input.shape(), permutation_shape}); in Compute() local131 const TensorShapes& output_matrix_shapes) { in ComputeTensorSlice()