Searched refs:unpack_op (Results 1 – 5 of 5) sorted by relevance
391 void RemoveUnpackOperator(const Operator& unpack_op, Model* model) { in RemoveUnpackOperator() argument392 for (const string& output_array_name : unpack_op.outputs) { in RemoveUnpackOperator()395 model->operators.erase(FindOperator(model, unpack_op)); in RemoveUnpackOperator()
2041 tensorflow::NodeDef* unpack_op = tensorflow_graph->add_node(); in ConvertUnpackOperator() local2042 unpack_op->set_op(op_name); in ConvertUnpackOperator()2043 unpack_op->set_name(src_op.outputs[0]); in ConvertUnpackOperator()2045 *unpack_op->add_input() = src_op.inputs[0]; in ConvertUnpackOperator()2048 (*unpack_op->mutable_attr())["T"].set_type(data_type); in ConvertUnpackOperator()2049 (*unpack_op->mutable_attr())["num"].set_i(src_op.num); in ConvertUnpackOperator()2050 (*unpack_op->mutable_attr())["axis"].set_i(src_op.axis); in ConvertUnpackOperator()
104 "unpack_op.cc",
289 tensorflow/core/kernels/unpack_op.cc
910 ":unpack_op",1182 name = "unpack_op",1183 prefix = "unpack_op",5562 "unpack_op.cc",