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 …]
1892 tensorflow::NodeDef* tile_op = tensorflow_graph->add_node(); in ConvertTileOperator() local1893 tile_op->set_op("Tile"); in ConvertTileOperator()1894 tile_op->set_name(src_op.outputs[0]); in ConvertTileOperator()1896 *tile_op->add_input() = src_op.inputs[0]; in ConvertTileOperator()1897 *tile_op->add_input() = src_op.inputs[1]; in ConvertTileOperator()1900 (*tile_op->mutable_attr())["T"].set_type(data_type); in ConvertTileOperator()1903 (*tile_op->mutable_attr())["Tmultiples"].set_type(multiples_data_type); in ConvertTileOperator()
1020 tile_op = array_ops.tile(labeled_tensor.tensor, multiples_list, name=scope)1026 return core.LabeledTensor(tile_op, new_axes)