| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_eager_test.py | 72 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
[all...] |
| topology_test.py | 186 a = input_layer_lib.Input(shape=(32,), name='input_a')
374 a = keras.layers.Input(shape=(32,), name='input_a')
435 a = keras.layers.Input(shape=(32,), name='input_a')
471 ['input_a', 'input_b'])
493 ['input_a', 'input_b'])
507 a = keras.layers.Input(shape=(32,), name='input_a')
570 a = keras.layers.Input(shape=(32,), name='input_a')
642 a = keras.layers.Input(shape=(32,), name='input_a')
694 a = keras.layers.Input(shape=(32,), name='input_a')
732 a = keras.layers.Input(shape=(10, 32), name='input_a')
[all...] |
| training_dataset_test.py | 244 input_a = keras.layers.Input(shape=(3,), name='input_1')
248 branch_a = [input_a, dense]
|
| training_test.py | 60 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])
137 'input_a:0': input_np,
185 input_a = keras.layers.Input(shape=(3,), name='input_a')
190 branch_a = [input_a, dense]
271 'input_a': input_a_np
[all...] |
| /external/vixl/examples/aarch64/ |
| simulated-runtime-calls.cc | 105 int32_t input_a = 1; local
107 simulator.WriteWRegister(0, input_a);
|
| /external/tensorflow/tensorflow/python/keras/ |
| models_test.py | 131 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...] |
| backend_test.py | 80 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),
[all...] |
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_constant_binary.cc | 43 const std::vector<int>& input_a,
47 DCHECK_EQ(input_a.size(), input_b.size());
49 DCHECK_EQ(input_a.size(), output_a->size());
50 DCHECK_EQ(selector.size(), input_a.size());
51 const int size = input_a.size();
54 (*output_a)[i] = input_a[i];
58 (*output_b)[i] = input_a[i];
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| xla_ops.cc | 518 shape_inference::ShapeHandle input_a = context->input(0);
522 if (context->RankKnown(input_a)) {
523 rank_a = context->Rank(input_a);
566 left_map[c] = context->Dim(input_a, left_map.size());
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| auto_parallel.h | 53 NodeDef* AddNodeDiv(const string& name, const string& input_a,
|
| auto_parallel.cc | 51 NodeDef* AutoParallel::AddNodeDiv(const string& name, const string& input_a,
56 node->add_input(input_a);
|
| /art/compiler/optimizing/ |
| instruction_simplifier_shared.cc | 39 HInstruction* input_a = input_other; local
80 mul->GetType(), op_kind, input_a, input_a, input_b, mul->GetDexPc());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_op_test.cc | 303 Tensor input_a(DT_FLOAT, {});
304 input_a.flat<float>().data()[0] = NODE_A_VAL2;
306 inputs.emplace_back(NAME_A, input_a);
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| multi_gpu_utils_test.py | 82 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/ops/ |
| list_ops.cc | 563 .Input("input_a: variant")
568 auto input_a = c->input(0);
570 TF_RETURN_IF_ERROR(c->Merge(input_a, input_b, &input_a));
571 c->set_output(0, input_a);
591 return errors::InvalidArgument("input_a.type != element_dtype: ",
|
| /external/tensorflow/tensorflow/python/tpu/ |
| feature_column_test.py | 225 input_a = np.array([
233 input_features = {'aaa': input_a, 'bbb': input_b}
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fractional_avg_pool_op_test.py | 301 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)
|
| fractional_max_pool_op_test.py | 298 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/keras/saving/ |
| saving_utils_test.py | 99 input_a = keras.layers.Input(shape=(input_dim,), name='input_a')
105 branch_a = [input_a, dense]
214 a = keras.layers.Input(shape=(3,), name='input_a')
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| recurrent_test.py | 858 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))
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| keras_test.py | 88 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])
179 'input_a': a_train,
186 'input_a': a_test
[all...] |
| keras_backward_compat_test.py | 66 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])
145 'input_a': a_train,
152 'input_a': a_test,
198 a = keras.layers.Input(shape=(3,), name='input_a')
564 {'input_a': input_a_np, 'input_b': input_b_np},
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| partially_decluster_pass_test.cc | 327 Output input_a = ops::Placeholder(s.WithOpName("input_a"), DT_INT32, local
334 Output shape = ops::Add(s.WithOpName("shape"), shape_of_mul, input_a);
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column_test.py | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner_test.cc | 366 Node* input_a = test::graph::Constant(&graph, a); local
369 .Input(input_a)
373 TF_ASSERT_OK(m.AddNode(input_a));
[all...] |
