Home
last modified time | relevance | path

Searched refs:dy_data_flat (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc126 auto dy_data_flat = dy_datas[i].flat<Y_T>(); in ComputeTheoreticalJacobianTranspose() local
127 dy_data_flat.setZero(); in ComputeTheoreticalJacobianTranspose()
146 auto dy_data_flat = dy_datas[y_idx].flat<Y_T>(); in ComputeTheoreticalJacobianTranspose() local
155 dy_data_flat(c) = unit; in ComputeTheoreticalJacobianTranspose()
171 dy_data_flat(c) = Y_T{0}; in ComputeTheoreticalJacobianTranspose()
/external/tensorflow/tensorflow/python/ops/
Dgradient_checker_v2.py164 dy_data_flat = dy_data.ravel().view(y_dtype.real_dtype.as_numpy_dtype)
169 dy_data_flat[col] = 1
172 dy_data_flat[col] = 0
Dgradient_checker.py100 dy_data_flat = dy_data.ravel().view(dy.dtype.real_dtype.as_numpy_dtype)
103 dy_data_flat[col] = 1
117 dy_data_flat[col] = 0