Searched refs:expand_dims_op (Results 1 – 2 of 2) sorted by relevance
499 (TFL_ExpandDimsOp:$expand_dims_op $input, $dim),500 (TFL_ReshapeOp $input, (ConstantOp (GetShape $expand_dims_op))),501 [(AnyStaticShapeTensor $expand_dims_op)]>;
1401 tensorflow::NodeDef* expand_dims_op = tensorflow_graph->add_node(); in ConvertExpandDimsOperator() local1402 expand_dims_op->set_op("ExpandDims"); in ConvertExpandDimsOperator()1403 expand_dims_op->set_name(src_op.outputs[0]); in ConvertExpandDimsOperator()1405 *expand_dims_op->add_input() = src_op.inputs[0]; in ConvertExpandDimsOperator()1406 *expand_dims_op->add_input() = src_op.inputs[1]; in ConvertExpandDimsOperator()1407 (*expand_dims_op->mutable_attr())["T"].set_type( in ConvertExpandDimsOperator()1409 (*expand_dims_op->mutable_attr())["Tdim"].set_type( in ConvertExpandDimsOperator()