Home
last modified time | relevance | path

Searched refs:split_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/
Dexport_tensorflow.cc993 auto* split_op = tensorflow_graph->add_node(); in ConvertSplitOperator() local
994 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() local
1297 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 …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
DBUILD79 "split_op.cc",
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_ra_init.cpp621 split_op(o); in split()
636 void ra_split::split_op(node* n) { in split_op() function in r600_sb::ra_split
Dsb_pass.h569 void split_op(node *n);
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt38 tensorflow/core/kernels/split_op.cc
/external/tensorflow/tensorflow/core/kernels/
DBUILD625 ":split_op",
829 name = "split_op",
831 prefix = "split_op",
3082 ":split_op",
4695 "split_op.cc",