Home
last modified time | relevance | path

Searched refs:x_data_flat (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc222 auto x_data_flat = (*x_datas)[x_idx].flat<X_T>(); in ComputeNumericJacobianTranspose() local
234 X_T v = x_data_flat(r); in ComputeNumericJacobianTranspose()
236 x_data_flat(r) = v + x_delta; in ComputeNumericJacobianTranspose()
240 x_data_flat(r) = v - x_delta; in ComputeNumericJacobianTranspose()
260 x_data_flat(r) = v; in ComputeNumericJacobianTranspose()
390 auto x_data_flat = x_datas[i].flat<X_T>(); in ComputeGradientError() local
391 x_data_flat.setRandom(); in ComputeGradientError()
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad_test.cc102 auto x_data_flat = x_data.flat<X_T>(); in TestCWiseGrad() local
103 for (int i = 0; i < x_data_flat.size(); ++i) { in TestCWiseGrad()
104 x_data_flat(i) = x_fn(i); in TestCWiseGrad()