Home
last modified time | relevance | path

Searched refs:batch_matmul_op_info (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1166 OpInfo batch_matmul_op_info = op_info; in GenerateBatchMatmulContextFromEinsum() local
1167 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()