Searched refs:expand_dims_node (Results 1 – 1 of 1) sorted by relevance
411 NodeDef expand_dims_node; in SparsifyGatherInternal() local412 expand_dims_node.set_op("ExpandDims"); in SparsifyGatherInternal()414 expand_dims_node.set_name(gather_node.name()); in SparsifyGatherInternal()415 SetNodeAttr("T", data_type, &expand_dims_node); in SparsifyGatherInternal()432 AddNodeInput(lookup_node.name(), &expand_dims_node); in SparsifyGatherInternal()434 AddNodeInput(dim_idx_node.name(), &expand_dims_node); in SparsifyGatherInternal()446 new_nodes->push_back(expand_dims_node); in SparsifyGatherInternal()