Searched refs:tile_op (Results 1 – 3 of 3) sorted by relevance
32 auto* tile_op = static_cast<TransposeOperator*>(tile_it->get()); in Run() local34 const auto& input_array = model->GetArray(tile_op->inputs[0]); in Run()35 const auto& multiples_array = model->GetArray(tile_op->inputs[1]); in Run()36 const auto& output_array = model->GetArray(tile_op->outputs[0]); in Run()64 LogName(*tile_op)); in Run()70 LogName(*tile_op), concat_axis); in Run()77 concat_op->inputs.push_back(tile_op->inputs[0]); in Run()80 concat_op->outputs = tile_op->outputs; in Run()83 if (IsDiscardableArray(*model, tile_op->inputs[1]) && in Run()84 CountOpsWithInput(*model, tile_op->inputs[1]) == 1) { in Run()[all …]
1926 tensorflow::NodeDef* tile_op = tensorflow_graph->add_node(); in ConvertTileOperator() local1927 tile_op->set_op("Tile"); in ConvertTileOperator()1928 tile_op->set_name(src_op.outputs[0]); in ConvertTileOperator()1930 *tile_op->add_input() = src_op.inputs[0]; in ConvertTileOperator()1931 *tile_op->add_input() = src_op.inputs[1]; in ConvertTileOperator()1934 (*tile_op->mutable_attr())["T"].set_type(data_type); in ConvertTileOperator()1937 (*tile_op->mutable_attr())["Tmultiples"].set_type(multiples_data_type); in ConvertTileOperator()
3010 auto tile_op = ops::Tile(scope.WithOpName("tile").WithDevice("/device:GPU:0"), in TEST_F() local3012 auto z = ops::Identity(scope.WithOpName("z"), tile_op); in TEST_F()