Searched refs:expand_dims_op (Results 1 – 1 of 1) sorted by relevance
1388 tensorflow::NodeDef* expand_dims_op = tensorflow_graph->add_node(); in ConvertExpandDimsOperator() local1389 expand_dims_op->set_op("ExpandDims"); in ConvertExpandDimsOperator()1390 expand_dims_op->set_name(src_op.outputs[0]); in ConvertExpandDimsOperator()1392 *expand_dims_op->add_input() = src_op.inputs[0]; in ConvertExpandDimsOperator()1393 *expand_dims_op->add_input() = src_op.inputs[1]; in ConvertExpandDimsOperator()1394 (*expand_dims_op->mutable_attr())["T"].set_type( in ConvertExpandDimsOperator()1396 (*expand_dims_op->mutable_attr())["Tdim"].set_type( in ConvertExpandDimsOperator()