| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_eager_test.py | 73 input_b = keras.layers.Input(shape=(3,), name='input_b')
79 [input_a, dense], [input_b, dense, dropout])
94 input_b = 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 | 187 b = input_layer_lib.Input(shape=(32,), name='input_b')
375 b = keras.layers.Input(shape=(32,), name='input_b')
436 b = keras.layers.Input(shape=(32,), name='input_b')
471 ['input_a', 'input_b'])
493 ['input_a', 'input_b'])
508 b = keras.layers.Input(shape=(32,), name='input_b')
571 b = keras.layers.Input(shape=(32,), name='input_b')
643 b = keras.layers.Input(shape=(32,), name='input_b')
695 b = keras.layers.Input(shape=(32,), name='input_b')
[all...] |
| training_dataset_test.py | 245 input_b = keras.layers.Input(shape=(3,), name='input_2')
249 branch_b = [input_b, dense, dropout]
|
| /external/vixl/examples/aarch64/ |
| simulated-runtime-calls.cc | 106 int32_t input_b = 2; local
108 simulator.WriteWRegister(1, input_b);
|
| /art/compiler/optimizing/ |
| instruction_simplifier_shared.cc | 40 HInstruction* input_b = nullptr; // Set to a non-null value if we found a pattern to optimize. local
49 input_b = input_binop->GetLeastConstantLeft();
60 input_b = input_binop->GetLeft();
68 input_b = input_binop->GetRight();
73 if (input_b == nullptr) {
80 mul->GetType(), op_kind, input_a, input_a, input_b, mul->GetDexPc());
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ops/ |
| xla_ops.cc | 519 shape_inference::ShapeHandle input_b = context->input(1);
527 if (context->RankKnown(input_b)) {
528 rank_b = context->Rank(input_b);
569 right_map[c] = context->Dim(input_b, right_map.size());
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| auto_parallel.h | 54 const string& input_b);
|
| auto_parallel.cc | 52 const string& input_b) {
57 node->add_input(input_b);
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| resolve_constant_binary.cc | 44 const std::vector<int>& input_b,
47 DCHECK_EQ(input_a.size(), input_b.size());
55 (*output_b)[i] = input_b[i];
57 (*output_a)[i] = input_b[i];
|
| /external/tensorflow/tensorflow/python/keras/ |
| models_test.py | 163 input_b = keras.Input(shape=(4,))
170 x_b = dense_1(input_b)
173 model = keras.models.Model([input_a, input_b], outputs)
185 input_b = keras.Input(shape=(4,), name='b')
187 model, input_tensors=[input_a, input_b])
199 input_b = keras.backend.variable(val_b)
200 new_model = clone_fn(model, input_tensors=[input_a, input_b])
|
| backend_test.py | 81 input_b = np.random.random(input_shape_b)
83 keras.backend.variable(input_b, dtype=dtype),
86 np_output = np_op(input_a.astype(dtype), input_b.astype(dtype),
[all...] |
| /external/libxcam/cl_kernel/ |
| kernel_defog_dcp.cl | 72 * input_b: B channel image2d_t as read only
84 __read_only image2d_t input_r, __read_only image2d_t input_g, __read_only image2d_t input_b,
100 in_b[0] = convert_float8(as_uchar8(convert_ushort4(read_imageui(input_b, sampler, (int2)(pos_x, pos_y)))));
101 in_b[1] = convert_float8(as_uchar8(convert_ushort4(read_imageui(input_b, sampler, (int2)(pos_x, pos_y + 1)))));
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| multi_gpu_utils_test.py | 83 input_b = keras.Input((input_dim_b,))
85 b = keras.layers.Dense(hidden_dim)(input_b)
89 model = keras.models.Model([input_a, input_b], [output_a, output_b])
|
| /external/tensorflow/tensorflow/python/tpu/ |
| feature_column_test.py | 229 input_b = np.array([
233 input_features = {'aaa': input_a, 'bbb': input_b}
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| partially_decluster_pass_test.cc | 329 Output input_b = ops::Placeholder(s.WithOpName("shape_b"), DT_FLOAT, local
331 Output mul = ops::Mul(s.WithOpName("mul"), input_b, input_b);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fractional_avg_pool_op_test.py | 307 input_b = np.zeros([4, 60, 60, 3])
308 actual, row_seq, col_seq = sess.run([p, r, c], {input_holder: input_b})
310 input_b, row_seq, col_seq, overlapping)
|
| fractional_max_pool_op_test.py | 304 input_b = np.zeros([4, 45, 45, 3])
305 actual, row_seq, col_seq = sess.run([p, r, c], {input_holder: input_b})
307 input_b, row_seq, col_seq, overlapping)
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| saving_utils_test.py | 100 input_b = keras.layers.Input(shape=(input_dim,), name='input_b')
106 branch_b = [input_b, dense, dense2, dropout]
215 b = keras.layers.Input(shape=(3,), name='input_b')
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| recurrent_test.py | 859 input_b = 10
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)),
908 input_b = 10
914 x = keras.Input((None, input_a, input_b))
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| keras_test.py | 89 input_b = keras.layers.Input(shape=(16,), name='input_b')
91 merged = keras.layers.concatenate([input_a, input_b], name='merge')
95 inputs=[input_a, input_b], outputs=[output_c, output_d])
101 input_b = keras.layers.Input(shape=(16,), name='input_b')
109 interm_b = dense(input_b)
114 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
180 'input_b': b_train,
187 'input_b': b_test
[all...] |
| keras_backward_compat_test.py | 67 input_b = keras.layers.Input(shape=(16,), name='input_b')
75 interm_b = dense(input_b)
80 inputs=[input_a, input_b, input_m], outputs=[output_c, output_d])
146 'input_b': b_train,
153 'input_b': b_test,
199 b = keras.layers.Input(shape=(5,), name='input_b')
564 {'input_a': input_a_np, 'input_b': input_b_np},
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| list_ops.cc | 564 .Input("input_b: variant")
569 auto input_b = c->input(1);
570 TF_RETURN_IF_ERROR(c->Merge(input_a, input_b, &input_a));
596 return errors::InvalidArgument("input_b.type != element_dtype: ",
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| feature_column_test.py | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner_test.cc | 367 Node* input_b = test::graph::Constant(&graph, b); local
370 .Input(input_b)
374 TF_ASSERT_OK(m.AddNode(input_b));
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| op_def_library_test.py | [all...] |
