Searched refs:dy_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 56 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
89 dy_size = _product(dy_shape) * dy_factor
99 dy_data = np.zeros(dy_shape, dtype=dy.dtype.as_numpy_dtype)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 157 const TensorShape dy_shape = ctx->InputShape(4); in Compile() local
159 ctx, final_shape == dy_shape, in Compile()
160 errors::InvalidArgument("shape of dy was ", dy_shape.DebugString(), in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 226 TensorShape dy_shape = context->input(4).shape(); in Compute() local
228 context, final_shape == dy_shape, in Compute()
229 errors::InvalidArgument("shape of dy was ", dy_shape.DebugString(), in Compute()
|