Home
last modified time | relevance | path

Searched defs:split_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc54 mlir::TF::SplitOp* split_op) { in CreateSplitOp()
177 for (auto split_op : split_ops_for_tiled_input) { in HandleTileShardedInputs() local
196 for (auto split_op : split_ops_for_tiled_input) in HandleTileShardedInputs() local
/external/mesa3d/src/gallium/drivers/r600/sb/
Dsb_ra_init.cpp640 void ra_split::split_op(node* n) { in split_op() function in r600_sb::ra_split
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1150 tensorflow::NodeDef* split_op = tensorflow_graph->add_node(); in ConvertSplitOperator() local
1567 tensorflow::NodeDef* split_op = tensorflow_graph->add_node(); in ConvertLstmCellOperator() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc3373 auto split_op = ops::Split( in TEST_F() local