Home
last modified time | relevance | path

Searched refs:input_layer (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/lite/testing/
Dgenerate_testspec.cc52 RandomEngine* engine, const std::vector<string>& input_layer, in GenerateInputValues() argument
56 input_values.resize(input_layer.size()); in GenerateInputValues()
57 for (int i = 0; i < input_layer.size(); i++) { in GenerateInputValues()
61 const auto& name = input_layer[i]; in GenerateInputValues()
100 const std::vector<string>& input_layer, in GenerateTestSpecFromRunner() argument
105 auto input_size = input_layer.size(); in GenerateTestSpecFromRunner()
117 const auto& name = input_layer[i]; in GenerateTestSpecFromRunner()
130 &random_engine, input_layer, input_layer_type, input_layer_shape); in GenerateTestSpecFromRunner()
171 const std::vector<string>& input_layer, in GenerateTestSpecFromTensorflowModel() argument
175 CHECK_EQ(input_layer.size(), input_layer_type.size()); in GenerateTestSpecFromTensorflowModel()
[all …]
Dtf_driver.cc106 TfDriver::TfDriver(const std::vector<string>& input_layer, in TfDriver() argument
110 : input_names_(input_layer), output_names_(output_layer) { in TfDriver()
111 CHECK_EQ(input_layer.size(), input_layer_type.size()); in TfDriver()
112 CHECK_EQ(input_layer.size(), input_layer_shape.size()); in TfDriver()
114 input_ids_.resize(input_layer.size()); in TfDriver()
115 input_tensors_.reserve(input_layer.size()); in TfDriver()
116 input_types_.resize(input_layer.size()); in TfDriver()
117 input_shapes_.resize(input_layer.size()); in TfDriver()
118 for (int i = 0; i < input_layer.size(); i++) { in TfDriver()
120 input_tensors_[input_layer[i]] = {}; in TfDriver()
[all …]
Dtflite_diff_flags.h34 string input_layer; in ParseTfliteDiffFlags() member
49 tensorflow::Flag("input_layer", &values.input_layer, in ParseTfliteDiffFlags()
78 values.input_layer.empty() || values.input_layer_type.empty() || in ParseTfliteDiffFlags()
100 Split<string>(values.input_layer, ","), in ParseTfliteDiffFlags()
Dtflite_model_test.bzl12 input_layer = "a,b,c,d",
24 input_layer,
39 input_layer: A list of input tensors to use in the test.
53 "--input_arrays=%s" % input_layer,
73 "--input_layer=%s" % input_layer,
Dgenerate_testspec.h44 const std::vector<string>& input_layer,
52 const std::vector<string>& input_layer,
Dtflite_diff_util.cc39 options.input_layer, options.input_layer_type, in SingleRunDiffTestWithProvidedRunner()
54 num_invocations, options.input_layer, options.input_layer_type, in RunDiffTest()
Dtflite_diff_util.h35 std::vector<string> input_layer; member
Dtf_driver.h38 explicit TfDriver(const std::vector<string>& input_layer,
/external/executorch/examples/models/llama/source_transformation/
Dprune_vocab.py95 input_layer = getattr(model, imput_layer_name)
97 input_layer, torch.nn.Embedding
99 original_shape = input_layer.weight.shape
101 weight_dtype = input_layer.weight.dtype
107 pruned_wt = input_layer.weight[token_id].detach()
/external/tensorflow/tensorflow/examples/label_image/
Dlabel_image.py73 input_layer = "input" variable
102 if args.input_layer:
103 input_layer = args.input_layer variable
115 input_name = "import/" + input_layer
Dmain.cc291 string input_layer = "input"; in main() local
304 Flag("input_layer", &input_layer, "name of input layer"), in main()
348 Status run_status = session->Run({{input_layer, resized_tensor}}, in main()
/external/tensorflow/tensorflow/python/keras/layers/
Dserialization.py24 from tensorflow.python.keras.engine import input_layer
42 ALL_MODULES = (base_layer, input_layer, advanced_activations, convolutional,
83 LOCAL.ALL_OBJECTS['Input'] = input_layer.Input
D__init__.py22 from tensorflow.python.keras.engine.input_layer import Input
23 from tensorflow.python.keras.engine.input_layer import InputLayer
/external/tensorflow/tensorflow/python/keras/engine/
Dsequential.py27 from tensorflow.python.keras.engine import input_layer
150 if layers and isinstance(layers[0], input_layer.InputLayer):
174 if isinstance(origin_layer, input_layer.InputLayer):
200 if isinstance(layer, input_layer.InputLayer):
207 x = input_layer.Input(
289 inputs = input_layer.Input(
Dinput_layer.py386 input_layer = InputLayer(**input_layer_config)
390 outputs = input_layer._inbound_nodes[0].outputs
Dfunctional.py29 from tensorflow.python.keras.engine import input_layer as input_layer_module
936 input_layer = input_t._keras_history[0]
937 if input_layer not in layers_depths:
938 layers_depths[input_layer] = 0
939 layer_indices[input_layer] = -1
940 nodes_depths[input_layer._inbound_nodes[0]] = 0
941 network_nodes.add(_make_node_key(input_layer.name, 0))
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2021-071.md13 input_layer = 2**60-1
19 shape=(input_layer, input_layer),
/external/tensorflow/tensorflow/tools/benchmark/
DREADME.md44 --input_layer="input:0" \
63 --input_layer="input:0" \
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1662 input_layer = fc.input_layer(
1669 input_layer_inputs = [tensor for tensor in input_layer.op.inputs[:-1]]
2600 input_layer = InputLayer(fc._numeric_column('a'))
2601 inputs = self.evaluate(input_layer(features))
2630 input_layer = InputLayer([embedding_column])
2633 inputs = input_layer(features)
2634 variables = input_layer.variables
2644 _ = input_layer(features)
2646 self.assertIs(variables[0], input_layer.variables[0])
2675 input_layer = InputLayer([embedding_column])
[all …]
Dfeature_column_v2_test.py1822 input_layer = fc_old.input_layer(
1829 input_layer_inputs = [tensor for tensor in input_layer.op.inputs[:-1]]
2293 input_layer = fc_old.InputLayer(fc.numeric_column('a'))
2294 inputs = self.evaluate(input_layer(features))
2324 input_layer = fc_old.InputLayer([embedding_column])
2327 inputs = input_layer(features)
2328 variables = input_layer.variables
2338 _ = input_layer(features)
2340 self.assertIs(variables[0], input_layer.variables[0])
2369 input_layer = fc_old.InputLayer([embedding_column])
[all …]
/external/tensorflow/tensorflow/python/keras/
D__init__.py27 from tensorflow.python.keras.engine.input_layer import Input
Dmodels.py28 from tensorflow.python.keras.engine.input_layer import Input
29 from tensorflow.python.keras.engine.input_layer import InputLayer
355 input_layer = input_tensor._keras_history.layer
356 cloned_model = Sequential(layers=[input_layer] + layers, name=model.name)
/external/armnn/python/pyarmnn/test/
Dtest_network.py313 input_layer = network.AddInputLayer(0, "input")
332 input_layer.GetOutputSlot(0).Connect(fully_connected.GetInputSlot(0))
337 input_layer.GetOutputSlot(0).SetTensorInfo(input_info)
/external/tensorflow/tensorflow/examples/multibox_detector/
Dmain.cc350 string input_layer = "ResizeBilinear"; in main() local
371 Flag("input_layer", &input_layer, "name of input layer"), in main()
420 session->Run({{input_layer, resized_tensor}}, in main()
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py68 input_layer = LazyLoader( variable
626 if isinstance(node, input_layer.InputLayer):
683 if not layers or not isinstance(layers[0], input_layer.InputLayer):
685 layers.insert(0, input_layer.InputLayer.from_config(
689 layers.insert(0, input_layer.InputLayer(
958 RevivedInputLayer, input_layer.InputLayer),

12