Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_eager_test.py72 input_a = keras.layers.Input(shape=(3,), name='input_a')
79 [input_a, dense], [input_b, dense, dropout])
93 input_a = array_ops.zeros(shape=(10, 3))
99 [input_a, input_b], [target_a, target_b],
105 [input_a, input_b], [target_a, target_b],
111 model.fit([input_a, input_b], [target_a, target_b],
113 validation_data=([input_a, input_b], [target_a, target_b]))
114 model.train_on_batch([input_a, input_b], [target_a, target_b])
115 model.predict([input_a, input_b], batch_size=5)
116 model.evaluate([input_a, input_b], [target_a, target_b],
[all …]
Dtopology_test.py958 input_a = input_layer_lib.Input(shape=(32,))
960 a = keras.layers.Dense(16)(input_a)
970 network = network_lib.Network([input_a, input_b], [a, c])
1030 input_a = input_layer_lib.Input(shape=(3,))
1032 output_a, output_b = MultiInputOutputLayer()([input_a, input_b])
1033 model = keras.Model([input_a, input_b], [output_a, output_b])
Dtraining_dataset_test.py244 input_a = keras.layers.Input(shape=(3,), name='input_1')
248 branch_a = [input_a, dense]
Dtraining_test.py60 input_a = keras.layers.Input(shape=(3,), name='input_a')
61 output_a = keras.layers.Dense(1, name='dense_1')(input_a)
62 output_b = keras.layers.Dense(1, name='dense_2')(input_a)
63 return keras.models.Model(input_a, [output_a, output_b])
185 input_a = keras.layers.Input(shape=(3,), name='input_a')
190 branch_a = [input_a, dense]
/external/tensorflow/tensorflow/python/keras/
Dmodels_test.py131 input_a = keras.Input(shape=(4,))
132 new_model = clone_fn(model, input_tensors=input_a)
141 input_a = keras.backend.variable(val_a)
142 new_model = clone_fn(model, input_tensors=input_a)
162 input_a = keras.Input(shape=(4,))
167 x_a = dense_1(input_a)
173 model = keras.models.Model([input_a, input_b], outputs)
184 input_a = keras.Input(shape=(4,), name='a')
187 model, input_tensors=[input_a, input_b])
198 input_a = keras.backend.variable(val_a)
[all …]
Dbackend_test.py80 input_a = np.random.random(input_shape_a)
82 keras_output = keras_op(keras.backend.variable(input_a, dtype=dtype),
86 np_output = np_op(input_a.astype(dtype), input_b.astype(dtype),
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc43 const std::vector<int>& input_a, in PairwiseVectorSelect() argument
47 DCHECK_EQ(input_a.size(), input_b.size()); in PairwiseVectorSelect()
49 DCHECK_EQ(input_a.size(), output_a->size()); in PairwiseVectorSelect()
50 DCHECK_EQ(selector.size(), input_a.size()); in PairwiseVectorSelect()
51 const int size = input_a.size(); in PairwiseVectorSelect()
54 (*output_a)[i] = input_a[i]; in PairwiseVectorSelect()
58 (*output_b)[i] = input_a[i]; in PairwiseVectorSelect()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc518 shape_inference::ShapeHandle input_a = context->input(0); in __anon3500048a0602() local
522 if (context->RankKnown(input_a)) { in __anon3500048a0602()
523 rank_a = context->Rank(input_a); in __anon3500048a0602()
566 left_map[c] = context->Dim(input_a, left_map.size()); in __anon3500048a0602()
/external/vixl/examples/aarch64/
Dsimulated-runtime-calls.cc105 int32_t input_a = 1; in main() local
107 simulator.WriteWRegister(0, input_a); in main()
/external/tensorflow/tensorflow/python/keras/utils/
Dmulti_gpu_utils_test.py82 input_a = keras.Input((input_dim_a,))
84 a = keras.layers.Dense(hidden_dim)(input_a)
89 model = keras.models.Model([input_a, input_b], [output_a, output_b])
/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/core/ops/
Dlist_ops.cc568 auto input_a = c->input(0); in __anon9187032f1402() local
570 TF_RETURN_IF_ERROR(c->Merge(input_a, input_b, &input_a)); in __anon9187032f1402()
571 c->set_output(0, input_a); in __anon9187032f1402()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_parallel.h53 NodeDef* AddNodeDiv(const string& name, const string& input_a,
Dauto_parallel.cc51 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a, in AddNodeDiv() argument
56 node->add_input(input_a); in AddNodeDiv()
/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.py225 input_a = np.array([
233 input_features = {'aaa': input_a, 'bbb': input_b}
/external/tensorflow/tensorflow/python/keras/layers/
Drecurrent_test.py858 input_a = 5
864 x = keras.Input((None, input_a, input_b))
882 np.zeros((batch, time_step, input_a, input_b)),
900 np.zeros((batch, time_step, input_a, input_b)),
907 input_a = 5
914 x = keras.Input((None, input_a, input_b))
925 np.zeros((batch, time_step, input_a, input_b)),
934 x = keras.Input((None, input_a, input_b))
1357 input_a = input_shape[-2]
1360 shape=(input_a, input_b, self.unit_a, self.unit_b),
/external/tensorflow/tensorflow/python/keras/saving/
Dsaving_utils_test.py99 input_a = keras.layers.Input(shape=(input_dim,), name='input_a')
105 branch_a = [input_a, dense]
/external/tensorflow/tensorflow/contrib/distribute/python/
Dkeras_test.py88 input_a = keras.layers.Input(shape=(16,), name='input_a')
91 merged = keras.layers.concatenate([input_a, input_b], name='merge')
95 inputs=[input_a, input_b], outputs=[output_c, output_d])
100 input_a = keras.layers.Input(shape=(16,), name='input_a')
105 interm_a = dense(input_a)
114 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
Dkeras_backward_compat_test.py66 input_a = keras.layers.Input(shape=(16,), name='input_a')
71 interm_a = dense(input_a)
80 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass_test.cc327 Output input_a = ops::Placeholder(s.WithOpName("input_a"), DT_INT32, in TEST() local
334 Output shape = ops::Add(s.WithOpName("shape"), shape_of_mul, input_a); in TEST()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py5738 input_a = np.array(
5745 'aaa': input_a,
5807 input_a = np.array([
5815 input_features = {'aaa': input_a, 'bbb': input_b}
5860 input_a = np.array(
5876 input_a_placeholder: input_a,
5918 input_a = np.array(
5947 categorical_column_a.name: input_a,
6009 input_a = np.array([
6041 categorical_column_a.name: input_a,
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner_test.cc366 Node* input_a = test::graph::Constant(&graph, a); in TEST_F() local
369 .Input(input_a) in TEST_F()
373 TF_ASSERT_OK(m.AddNode(input_a)); in TEST_F()
/external/tensorflow/tensorflow/python/framework/
Dop_def_library_test.py1311 input_a = self._lib.apply_op("RefOut", T=dtypes.int32, name="t")
1313 op = self._lib.apply_op("TwoRefsIn", a=input_a, b=input_b, name="v")

12