Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/testing/
Dgenerate_testspec.cc41 RandomEngine* engine, const std::vector<string>& input_layer, in GenerateInputValues() argument
45 input_values.resize(input_layer.size()); in GenerateInputValues()
46 for (int i = 0; i < input_layer.size(); i++) { in GenerateInputValues()
88 const std::vector<string>& input_layer, in GenerateTestSpecFromRunner() argument
105 &random_engine, input_layer, input_layer_type, input_layer_shape); in GenerateTestSpecFromRunner()
154 const std::vector<string>& input_layer, in GenerateTestSpecFromTensorflowModel() argument
158 CHECK_EQ(input_layer.size(), input_layer_type.size()); in GenerateTestSpecFromTensorflowModel()
159 CHECK_EQ(input_layer.size(), input_layer_shape.size()); in GenerateTestSpecFromTensorflowModel()
162 TfDriver runner(input_layer, input_layer_type, input_layer_shape, in GenerateTestSpecFromTensorflowModel()
176 return GenerateTestSpecFromRunner(stream, num_invocations, input_layer, in GenerateTestSpecFromTensorflowModel()
[all …]
Dtf_driver.cc105 TfDriver::TfDriver(const std::vector<string>& input_layer, in TfDriver() argument
109 : input_names_(input_layer), output_names_(output_layer) { in TfDriver()
110 CHECK_EQ(input_layer.size(), input_layer_type.size()); in TfDriver()
111 CHECK_EQ(input_layer.size(), input_layer_shape.size()); in TfDriver()
113 input_ids_.resize(input_layer.size()); in TfDriver()
114 input_tensors_.reserve(input_layer.size()); in TfDriver()
115 input_types_.resize(input_layer.size()); in TfDriver()
116 input_shapes_.resize(input_layer.size()); in TfDriver()
117 for (int i = 0; i < input_layer.size(); i++) { in TfDriver()
119 input_tensors_[input_layer[i]] = {}; in TfDriver()
Dtflite_diff_flags.h33 string input_layer; in ParseTfliteDiffFlags() member
48 tensorflow::Flag("input_layer", &values.input_layer, in ParseTfliteDiffFlags()
77 values.input_layer.empty() || values.input_layer_type.empty() || in ParseTfliteDiffFlags()
99 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.cc38 options.input_layer, options.input_layer_type, in SingleRunDiffTestWithProvidedRunner()
53 num_invocations, options.input_layer, options.input_layer_type, in RunDiffTest()
Dtflite_diff_util.h34 std::vector<string> input_layer; member
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dcategory_crossing_test.py30 from tensorflow.python.keras.engine import input_layer
109 inp_0_t = input_layer.Input(shape=(1,), sparse=True, dtype=dtypes.string)
110 inp_1_t = input_layer.Input(shape=(1,), sparse=True, dtype=dtypes.string)
111 inp_2_t = input_layer.Input(shape=(1,), sparse=True, dtype=dtypes.string)
131 inp_0_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
132 inp_1_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
150 inp_0_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
151 inp_1_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
163 inp_0_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
164 inp_1_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
[all …]
Dhashing_test.py30 from tensorflow.python.keras.engine import input_layer
181 inp_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
211 inp_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.int64)
226 inp_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.string)
249 inp_t = input_layer.Input(shape=(None,), ragged=True, dtype=dtypes.int64)
/external/tensorflow/tensorflow/python/keras/feature_column/
Dsequence_feature_column_test.py138 input_layer, sequence_length = sequence_input_layer({
149 self.assertAllEqual(expected_input_layer, self.evaluate(input_layer))
226 input_layer, sequence_length = sequence_input_layer({
237 input_layer.eval(session=sess))
334 input_layer, sequence_length = sequence_input_layer({
337 self.assertAllEqual(expected_input_layer, self.evaluate(input_layer))
393 input_layer, sequence_length = sequence_input_layer({'aaa': sparse_input})
395 self.assertAllEqual(expected_input_layer, self.evaluate(input_layer))
436 input_layer, sequence_length = sequence_input_layer({'aaa': sparse_input})
438 self.assertAllEqual(expected_input_layer, self.evaluate(input_layer))
[all …]
/external/tensorflow/tensorflow/examples/label_image/
Dlabel_image.py82 input_layer = "input" variable
111 if args.input_layer:
112 input_layer = args.input_layer variable
124 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/engine/
Dnode_test.py111 input_layer = DummyLayer()
114 node_a = node_module.Node(layer=input_layer, outputs=a)
120 node_c = node_module.Node(layer=input_layer, outputs=arg_3)
124 node_d = node_module.Node(layer=input_layer, outputs=kwarg_y)
149 self.assertLen(input_layer._outbound_nodes, 3)
Dbase_layer_test.py45 from tensorflow.python.keras.engine import input_layer
175 inputs = input_layer.Input((3,))
180 inputs = input_layer.Input((3,))
250 inputs = input_layer.Input((3,))
255 inputs = input_layer.Input((1,))
281 _ = sequential.Sequential([input_layer.Input(shape=(4,))] + layer)
369 inputs = input_layer.Input(shape=(1,))
405 inputs = input_layer.Input(shape=(1,))
484 inputs = input_layer.Input(shape=[2])
510 inputs = input_layer.Input(shape=[2])
[all …]
Dtraining_gpu_test.py26 from tensorflow.python.keras.engine import input_layer
100 inputs = input_layer.Input(shape=(3, 3, 1))
111 inputs = input_layer.Input(shape=(1, 3, 3))
Dsequential.py31 from tensorflow.python.keras.engine import input_layer
155 if layers and isinstance(layers[0], input_layer.InputLayer):
179 if isinstance(origin_layer, input_layer.InputLayer):
205 if isinstance(layer, input_layer.InputLayer):
212 x = input_layer.Input(
294 inputs = input_layer.Input(
/external/tensorflow/tensorflow/python/keras/utils/
Dcomposite_tensor_support_test.py36 from tensorflow.python.keras.engine import input_layer
345 model_input = input_layer.Input(
382 model_input = input_layer.Input(shape=(3,), sparse=True, dtype=dtypes.int64)
402 model_input = input_layer.Input(shape=(3,), sparse=True, dtype=dtypes.int64)
431 model_input = input_layer.Input(
462 model_input = input_layer.Input(
506 model_input = input_layer.Input(
554 model_input = input_layer.Input(
583 model_input = input_layer.Input(
616 model_input = input_layer.Input(
[all …]
/external/tensorflow/tensorflow/python/keras/tests/
Dserialization_util_test.py26 from tensorflow.python.keras.engine import input_layer
57 x = input_layer.Input(shape=[3])
/external/tensorflow/tensorflow/python/keras/premade/
Dwide_deep_test.py26 from tensorflow.python.keras.engine import input_layer
98 inp = input_layer.Input(shape=(1,), name='linear')
140 linear_input = input_layer.Input(shape=(3,), name='linear')
141 dnn_input = input_layer.Input(shape=(5,), name='dnn')
144 input_b = input_layer.Input(shape=(1,), name='b')
Dlinear_test.py33 from tensorflow.python.keras.engine import input_layer
83 input_a = input_layer.Input(shape=(1,), name='a')
85 input_b = input_layer.Input(shape=(1,), name='b')
/external/tensorflow/tensorflow/tools/benchmark/
DREADME.md44 --input_layer="input:0" \
63 --input_layer="input:0" \
/external/tensorflow/tensorflow/python/keras/layers/
Dserialization.py28 from tensorflow.python.keras.engine import input_layer
67 ALL_MODULES = (base_layer, input_layer, advanced_activations, convolutional,
126 LOCAL.ALL_OBJECTS['Input'] = input_layer.Input
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py1683 input_layer = fc.input_layer(
1690 input_layer_inputs = [tensor for tensor in input_layer.op.inputs[:-1]]
2611 input_layer = InputLayer(fc._numeric_column('a'))
2612 inputs = self.evaluate(input_layer(features))
2641 input_layer = InputLayer([embedding_column])
2644 inputs = input_layer(features)
2645 variables = input_layer.variables
2655 _ = input_layer(features)
2657 self.assertIs(variables[0], input_layer.variables[0])
2686 input_layer = InputLayer([embedding_column])
[all …]
Dfeature_column_v2_test.py1826 input_layer = fc_old.input_layer(
1833 input_layer_inputs = [tensor for tensor in input_layer.op.inputs[:-1]]
2297 input_layer = fc_old.InputLayer(fc.numeric_column('a'))
2298 inputs = self.evaluate(input_layer(features))
2328 input_layer = fc_old.InputLayer([embedding_column])
2331 inputs = input_layer(features)
2332 variables = input_layer.variables
2342 _ = input_layer(features)
2344 self.assertIs(variables[0], input_layer.variables[0])
2373 input_layer = fc_old.InputLayer([embedding_column])
[all …]
/external/tensorflow/tensorflow/python/keras/
D__init__.py32 from tensorflow.python.keras.engine.input_layer import Input

123