Searched refs:dy_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradient_checker.py | 57 def _compute_theoretical_jacobian(x, x_shape, x_data, dy, dy_shape, dx, argument
90 dy_size = _product(dy_shape) * dy_factor
100 dy_data = np.zeros(dy_shape, dtype=dy.dtype.as_numpy_dtype)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 237 TensorShape dy_shape = context->input(4).shape(); in Compute() local
239 context, final_shape == dy_shape, in Compute()
240 errors::InvalidArgument("shape of dy was ", dy_shape.DebugString(), in Compute()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 512 const TensorShape dy_shape = ctx->InputShape(4); in Compile() local
514 ctx, final_shape == dy_shape, in Compile()
515 errors::InvalidArgument("shape of dy was ", dy_shape.DebugString(), in Compile()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 1574 dy_shape = array_ops.shape(dy)
1575 dx = gen_nn_ops.conv3d_backprop_input_v2(dy_shape, w, dy, strides, 'SAME')
|