Searched refs:output_matrix_shapes (Results 1 – 3 of 3) sorted by relevance
97 TensorShapes output_matrix_shapes; in Compute() local100 &output_matrix_shapes); in Compute()105 &output_matrix_shapes, context](int64 begin, int64 end) { in Compute()108 output_matrix_shapes); in Compute()164 TensorShapes* output_matrix_shapes) { in PrepareOutputs() argument166 *output_matrix_shapes = GetOutputMatrixShapes(input_matrix_shapes); in PrepareOutputs()167 const int num_outputs = output_matrix_shapes->size(); in PrepareOutputs()186 output_matrix_shapes->at(output_idx); in PrepareOutputs()223 const TensorShapes& output_matrix_shapes) { in ComputeTensorSlice() argument235 for (size_t i = 0; i < output_matrix_shapes.size(); ++i) { in ComputeTensorSlice()[all …]
96 TensorShapes output_matrix_shapes({input.shape(), permutation_shape}); in Compute() local116 &output_matrix_shapes, context](int64 begin, int64 end) { in Compute()119 output_matrix_shapes); in Compute()131 const TensorShapes& output_matrix_shapes) { in ComputeTensorSlice() argument
177 const TensorShapes& output_matrix_shapes);186 TensorShapes* output_matrix_shapes);