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