Home
last modified time | relevance | path

Searched refs:y_neg (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc65 Node* y_neg = test::graph::Unary(&graph, "Neg", y); in TEST() local
66 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST()
89 std::vector<string> target_nodes = {y_neg->name()}; in TEST()
104 if (node->name() == y->name() || node->name() == y_neg->name()) { in TEST()
189 Node* y_neg = test::graph::Unary(&graph, "Neg", y); in TestHWAccelerator() local
190 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:0"); in TestHWAccelerator()
209 std::vector<string> target_nodes = {y_neg->name()}; in TestHWAccelerator()
231 if (node->name() == y->name() || node->name() == y_neg->name()) { in TestHWAccelerator()
274 Node* y_neg = test::graph::Unary(&graph, "Neg", y); in TEST() local
275 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in TEST()
[all …]
Ddirect_session_test.cc104 Node* y_neg = test::graph::Unary(&graph, "Neg", y); in Initialize() local
105 y_neg_ = y_neg->name(); in Initialize()
106 y_neg->set_assigned_device_name("/job:localhost/replica:0/task:0/cpu:1"); in Initialize()
108 Node* z = test::graph::Unary(&graph, "Identity", y_neg); in Initialize()
/external/tensorflow/tensorflow/core/platform/
Ddevice_tracer_test.cc78 Node* y_neg = test::graph::Unary(&graph, "Neg", i); in Initialize() local
79 y_neg_ = y_neg->name(); in Initialize()
80 y_neg->set_assigned_device_name( in Initialize()
/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc241 std::vector<Tensor> y_neg; in ComputeNumericJacobianTranspose() local
242 TF_RETURN_IF_ERROR(EvaluateGraph(&session, xs, ys, x_datas, &y_neg)); in ComputeNumericJacobianTranspose()
248 auto y_neg_flat = y_neg[y_idx].flat<Y_T>(); in ComputeNumericJacobianTranspose()
/external/tensorflow/tensorflow/python/ops/
Dgradient_checker_v2.py250 y_neg = _to_numpy(f(*xs))
252 diff = (y_pos - y_neg) / scale
Dgradient_checker.py187 y_neg = y.eval(feed_dict=_extra_feeds(extra_feed_dict, {x: x_neg}))
188 diff = (y_pos - y_neg) / scale
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py1129 y_neg = sess.run(y, feed_dict=y_neg_feed_dict)
1130 numeric_grad[i] = (y_pos - y_neg) / (2 * delta)