Searched refs:batch_matmul_op_info (Results 1 – 1 of 1) sorted by relevance
1166 OpInfo batch_matmul_op_info = op_info; in GenerateBatchMatmulContextFromEinsum() local1167 batch_matmul_op_info.mutable_inputs()->Clear(); in GenerateBatchMatmulContextFromEinsum()1168 batch_matmul_op_info.set_op("BatchMatMul"); in GenerateBatchMatmulContextFromEinsum()1172 (*batch_matmul_op_info.mutable_attr())["transpose_a"] = transpose_attribute; in GenerateBatchMatmulContextFromEinsum()1173 (*batch_matmul_op_info.mutable_attr())["transpose_b"] = transpose_attribute; in GenerateBatchMatmulContextFromEinsum()1175 OpInfo::TensorProperties* a_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum()1179 OpInfo::TensorProperties* b_matrix = batch_matmul_op_info.add_inputs(); in GenerateBatchMatmulContextFromEinsum()1230 batch_matmul_context->op_info = batch_matmul_op_info; in GenerateBatchMatmulContextFromEinsum()