| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 141 TensorShape grad_shape = ctx->InputShape(3); in Compile() local
142 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
145 var_shape.DebugString(), " ", grad_shape.DebugString())); in Compile()
199 TensorShape grad_shape = ctx->InputShape(3); in Compile() local
200 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
203 var_shape.DebugString(), " ", grad_shape.DebugString())); in Compile()
255 TensorShape grad_shape = ctx->InputShape(3); in Compile() local
256 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
259 var_shape.DebugString(), " ", grad_shape.DebugString())); in Compile()
304 TensorShape grad_shape = ctx->InputShape(5); in Compile() local
[all …]
|
| D | image_resize_ops.cc | 619 TensorShape grad_shape = ctx->InputShape(0); in Compile() local
620 OP_REQUIRES(ctx, grad_shape.dims() == 4, in Compile()
622 grad_shape.DebugString())); in Compile()
623 const int64 grad_batch = grad_shape.dim_size(0); in Compile()
624 const std::vector<int64> grad_size = {grad_shape.dim_size(1), in Compile()
625 grad_shape.dim_size(2)}; in Compile()
626 const int64 grad_channels = grad_shape.dim_size(3); in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | shape_ops_test.py | 569 grad_shape = [4, 4]
570 grad_inp = np.random.rand(*grad_shape).astype("f")
572 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
585 grad_shape = [4, 4]
586 grad_inp = np.random.rand(*grad_shape).astype("f")
588 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
605 grad_shape = [4, 4]
606 grad_inp = np.random.rand(*grad_shape).astype("f")
608 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
620 grad_shape = [4, 4]
[all …]
|
| D | sparse_conditional_accumulator_test.py | 516 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()
524 q.apply_grad(grad_indices=[0], grad_values=1.0, grad_shape=[]).run()
531 q.apply_grad(grad_indices=[0], grad_values=[1.0], grad_shape=[]).run()
547 grad_shape=[2, 2]).run()
556 grad_shape=[2, 3, 2]).run()
666 grad_shape=constant_op.constant(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| D | tensorflow.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.sparse.-sparse-conditional-accumulator.pbtxt | 24 …argspec: "args=[\'self\', \'grad_indices\', \'grad_values\', \'grad_shape\', \'local_step\', \'nam…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | data_flow_ops.py | 1385 grad_shape=grad.dense_shape,
1392 grad_shape=None, argument
1433 [] if grad_shape is None else grad_shape, _dtypes.int64),
1434 has_known_shape=(grad_shape is not None),
|
| D | array_grad.py | 352 grad_shape = array_ops.shape(grad)
354 batch_shape = array_ops.slice(grad_shape, [0], [grad_rank - 2])
355 matrix_shape = array_ops.slice(grad_shape, [grad_rank - 2], [2])
|
| D | math_grad.py | 894 grad_shape = grad._shape_tuple()
896 return (x_shape == y_shape and x_shape == grad_shape and
|
| D | control_flow_ops.py | 1289 grad_shape = array_ops.shape_internal(enter_grad, optimize=False)
1290 grad_val = array_ops.zeros(grad_shape)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 2226 grad_shape = array_ops.shape(grad)
2227 last_dim_shape = grad_shape[-1]
2228 first_dim_shape = grad_shape[0]
|