Searched defs:output_matrix_shapes (Results 1 – 2 of 2) sorted by relevance
97 TensorShapes output_matrix_shapes; in Compute() local164 TensorShapes* output_matrix_shapes) { in PrepareOutputs()223 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()