Searched refs:input_layers (Results 1 – 4 of 4) sorted by relevance
444 std::vector<string> input_layers = str_util::Split(input_layer_string, ','); in Main() local453 if ((input_layers.size() != input_layer_shapes.size()) || in Main()454 (input_layers.size() != input_layer_types.size())) { in Main()460 << input_layers.size() << " items)"; in Main()467 const size_t inputs_count = input_layers.size(); in Main()555 input.name = input_layers[n]; in Main()
70 std::vector<string> input_layers; in PrintBenchmarkUsage() local74 input_layers.push_back(node->name()); in PrintBenchmarkUsage()106 string input_layer_value = absl::StrJoin(input_layers, ","); in PrintBenchmarkUsage()
217 def _clone_layers_and_model_config(model, input_layers, layer_fn): argument237 if layer in input_layers:238 created_layers[layer.name] = input_layers[layer]
1293 input_layers = tf_utils.convert_inner_node_data(1295 for layer_data in nest.flatten(input_layers):1313 input_tensors = nest.pack_sequence_as(input_layers, input_tensors)