Searched defs:a_input_shape (Results 1 – 1 of 1) sorted by relevance
290 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in IsEinsumCorrectlyFormed() local1155 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in GenerateBatchMatmulContextFromEinsum() local1273 TensorShapeProto a_input_shape = MaybeGetMinimumShape( in CountBatchMatMulOperations() local