Home
last modified time | relevance | path

Searched refs:input_a (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager_test.py74 input_a = keras.layers.Input(shape=(3,), name='input_a')
81 [input_a, dense], [input_b, dense, dropout])
95 input_a = array_ops.zeros(shape=(10, 3))
101 [input_a, input_b], [target_a, target_b],
107 [input_a, input_b], [target_a, target_b],
113 model.fit([input_a, input_b], [target_a, target_b],
115 validation_data=([input_a, input_b], [target_a, target_b]))
116 model.train_on_batch([input_a, input_b], [target_a, target_b])
117 model.predict([input_a, input_b], batch_size=5)
118 model.evaluate([input_a, input_b], [target_a, target_b],
[all …]
Dtraining_dataset_test.py158 input_a = keras.layers.Input(shape=(3,), name='input_1')
162 branch_a = [input_a, dense]
/external/tensorflow/tensorflow/python/keras/premade/
Dlinear_test.py57 input_a = np.random.uniform(low=-5, high=5, size=(64, 1))
59 output = .3 * input_a + .2 * input_b
61 model.fit([input_a, input_b], output, epochs=5)
65 input_a = np.random.uniform(low=-5, high=5, size=(64, 1))
67 output = .3 * input_a + .2 * input_b
72 model.fit({'c': input_a, 'b': input_b}, output, epochs=5)
76 input_a = np.random.uniform(low=-5, high=5, size=(64, 1))
78 output = .3 * input_a + .2 * input_b
80 model.fit({'a': input_a, 'b': input_b}, output, epochs=5)
83 input_a = input_layer.Input(shape=(1,), name='a')
[all …]
/external/tensorflow/tensorflow/python/keras/
Dmodels_test.py134 input_a = keras.Input(shape=(4,))
135 new_model = clone_fn(model, input_tensors=input_a)
146 input_a = keras.backend.variable(val_a)
147 new_model = clone_fn(model, input_tensors=input_a)
170 input_a = keras.Input(shape=(4,))
175 x_a = dense_1(input_a)
181 model = keras.models.Model([input_a, input_b], outputs)
194 input_a = keras.Input(shape=(4,), name='a')
197 model, input_tensors=[input_a, input_b])
210 input_a = keras.backend.variable(val_a)
[all …]
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc44 const std::vector<int>& input_a, in PairwiseVectorSelect() argument
48 DCHECK_EQ(input_a.size(), input_b.size()); in PairwiseVectorSelect()
50 DCHECK_EQ(input_a.size(), output_a->size()); in PairwiseVectorSelect()
51 DCHECK_EQ(selector.size(), input_a.size()); in PairwiseVectorSelect()
52 const int size = input_a.size(); in PairwiseVectorSelect()
55 (*output_a)[i] = input_a[i]; in PairwiseVectorSelect()
59 (*output_b)[i] = input_a[i]; in PairwiseVectorSelect()
/external/vixl/examples/aarch64/
Dsimulated-runtime-calls.cc108 int32_t input_a = 1; in main() local
110 simulator.WriteWRegister(0, input_a); in main()
/external/tensorflow/tensorflow/compiler/jit/
Dintroduce_floating_point_jitter_pass_test.cc40 Output input_a = ops::Placeholder(root.WithOpName("input_a"), DT_FLOAT); in TEST() local
43 Output sigmoid_a = ops::Sigmoid(root.WithOpName("sigmoid_a"), input_a); in TEST()
82 Output input_a = ops::Placeholder(root.WithOpName("input_a"), DT_FLOAT); in TEST() local
85 Output sigmoid_a = ops::Sigmoid(root.WithOpName("sigmoid_a"), input_a); in TEST()
Dpartially_decluster_pass_test.cc324 Output input_a = ops::Placeholder(s.WithOpName("input_a"), DT_INT32, in TEST() local
331 Output shape = ops::Add(s.WithOpName("shape"), shape_of_mul, input_a); in TEST()
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
Dsaved_model_api_test.cc118 TFE_TensorHandle* input_a = TestScalarTensorHandle(ctx, 2.0f); in TEST_P() local
120 compute_fn_inputs.push_back(input_a); in TEST_P()
145 TFE_DeleteTensorHandle(input_a); in TEST_P()
257 TFE_TensorHandle* input_a = TestScalarTensorHandle(ctx, 2.0f); in TEST_P() local
259 compute_fn_inputs.push_back(input_a); in TEST_P()
285 TFE_DeleteTensorHandle(input_a); in TEST_P()
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTensorListConcatLists.pbtxt4 name: "input_a"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTensorListConcatLists.pbtxt4 name: "input_a"
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_op_test.cc303 Tensor input_a(DT_FLOAT, {}); in TEST() local
304 input_a.flat<float>().data()[0] = NODE_A_VAL2; in TEST()
306 inputs.emplace_back(NAME_A, input_a); in TEST()
/external/tensorflow/tensorflow/python/keras/utils/
Dgeneric_utils_test.py376 input_a = list([1, 2, 3])
378 keras.utils.generic_utils.slice_arrays(input_a, start=0),
381 keras.utils.generic_utils.slice_arrays(input_a, stop=3),
384 keras.utils.generic_utils.slice_arrays(input_a, start=0, stop=1),
Dmulti_gpu_utils_test.py99 input_a = keras.Input((input_dim_a,))
101 a = keras.layers.Dense(hidden_dim)(input_a)
106 model = keras.models.Model([input_a, input_b], [output_a, output_b])
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_parallel.h55 NodeDef* AddNodeDiv(const string& name, const string& input_a,
Dauto_parallel.cc52 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a, in AddNodeDiv() argument
57 node->add_input(input_a); in AddNodeDiv()
/external/tensorflow/tensorflow/core/ops/
Dlist_ops.cc596 auto input_a = c->input(0); in __anonabedecb61402() local
598 TF_RETURN_IF_ERROR(c->Merge(input_a, input_b, &input_a)); in __anonabedecb61402()
599 c->set_output(0, input_a); in __anonabedecb61402()
/external/tensorflow/tensorflow/python/kernel_tests/
Dfractional_avg_pool_op_test.py301 input_a = np.zeros([3, 32, 32, 3])
302 actual, row_seq, col_seq = sess.run([p, r, c], {input_holder: input_a})
304 input_a, row_seq, col_seq, overlapping)
Dfractional_max_pool_op_test.py298 input_a = np.zeros([3, 32, 32, 3])
299 actual, row_seq, col_seq = sess.run([p, r, c], {input_holder: input_a})
301 input_a, row_seq, col_seq, overlapping)
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_test.py250 input_a = np.array([
258 input_features = {'aaa': input_a, 'bbb': input_b}
Dfeature_column_v2_test.py360 input_a = sparse_tensor.SparseTensorValue(
373 input_features = {'aaa': input_a, 'bbb': input_b}
/external/tensorflow/tensorflow/python/ops/
Dsparse_ops_test.py293 input_a = array_ops.sparse_placeholder(dtypes.float32, shape=(2, 1))
296 result = sparse_ops.sparse_concat_v2(axis=1, sp_inputs=[input_a, input_b])
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_test.py944 input_a = 5
950 x = keras.Input((None, input_a, input_b))
967 np.zeros((batch, time_step, input_a, input_b)),
985 np.zeros((batch, time_step, input_a, input_b)),
992 input_a = 5
999 x = keras.Input((None, input_a, input_b))
1010 np.zeros((batch, time_step, input_a, input_b)),
1019 x = keras.Input((None, input_a, input_b))
1804 input_a = input_shape[-2]
1807 shape=(input_a, input_b, self.unit_a, self.unit_b),
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py118 input_a = keras.layers.Input(shape=(input_dim,), name='input_a')
124 branch_a = [input_a, dense]
/external/tensorflow/tensorflow/python/keras/distribute/
Ddistribute_strategy_test.py113 input_a = keras.layers.Input(shape=(16,), name='input_a')
116 merged = keras.layers.concatenate([input_a, input_b], name='merge')
120 inputs=[input_a, input_b], outputs=[output_c, output_d])
1514 input_a = keras.layers.Input(shape=(16,), name='z_input_sorted_last')
1516 intermediate_a = keras.layers.Dense(10)(input_a)
1520 return input_a, input_b, output
1532 input_a, input_b, output = _create_model_input_output_tensors()
1540 inputs=[input_a, input_b], outputs=output)
1546 input_a, input_b, output = _create_model_input_output_tensors()
1549 'z_input_sorted_last': input_a,

12