Searched refs:split_op (Results 1 – 6 of 6) sorted by relevance
993 auto* split_op = tensorflow_graph->add_node(); in ConvertSplitOperator() local994 split_op->set_op("Split"); in ConvertSplitOperator()995 split_op->set_name(src_op.outputs[0]); in ConvertSplitOperator()997 *split_op->add_input() = input; in ConvertSplitOperator()999 (*split_op->mutable_attr())["T"].set_type(DT_FLOAT); in ConvertSplitOperator()1000 (*split_op->mutable_attr())["num_split"].set_i(src_op.num_split); in ConvertSplitOperator()1296 auto* split_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local1297 split_op->set_op("Split"); in ConvertLstmCellOperator()1298 split_op->set_name(split_output); in ConvertLstmCellOperator()1299 *split_op->add_input() = split_dim_output; in ConvertLstmCellOperator()[all …]
79 "split_op.cc",
621 split_op(o); in split()636 void ra_split::split_op(node* n) { in split_op() function in r600_sb::ra_split
569 void split_op(node *n);
38 tensorflow/core/kernels/split_op.cc
625 ":split_op",829 name = "split_op",831 prefix = "split_op",3082 ":split_op",4695 "split_op.cc",