| /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()
257 TensorShape grad_shape = ctx->InputShape(3); in Compile() local
258 OP_REQUIRES(ctx, var_shape.IsSameSize(grad_shape), in Compile()
261 var_shape.DebugString(), " ", grad_shape.DebugString())); in Compile()
311 TensorShape grad_shape = ctx->InputShape(4); in Compile() local
[all …]
|
| D | image_resize_ops.cc | 655 TensorShape grad_shape = ctx->InputShape(0); in Compile() local
656 OP_REQUIRES(ctx, grad_shape.dims() == 4, in Compile()
658 grad_shape.DebugString())); in Compile()
659 const int64 grad_batch = grad_shape.dim_size(0); in Compile()
660 const std::vector<int64> grad_size = {grad_shape.dim_size(1), in Compile()
661 grad_shape.dim_size(2)}; in Compile()
662 const int64 grad_channels = grad_shape.dim_size(3); in Compile()
|
| D | strided_slice_op.cc | 414 xla::Shape grad_shape = ctx->InputXlaShape(4).ValueOrDie(); in CompileAsDynamicUpdateSlice() local
415 VLOG(1) << "xla grad shape" << grad_shape; in CompileAsDynamicUpdateSlice()
430 for (int64 i = 0; i < grad_shape.rank(); ++i) { in CompileAsDynamicUpdateSlice()
436 grad_shape.dimensions(i)); in CompileAsDynamicUpdateSlice()
566 xla::Shape grad_shape = ctx->builder()->GetShape(grad).ValueOrDie(); in Compile() local
571 DCHECK_EQ(grad_shape.rank(), input_shape.dims()); in Compile()
573 DCHECK_EQ(grad_shape.dimensions(dim), input_shape.dim_size(dim)); in Compile()
581 } else if (grad_shape.is_dynamic_dimension(dim)) { in Compile()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | shape_ops_test.py | 573 grad_shape = [4, 4]
574 grad_inp = np.random.rand(*grad_shape).astype("f")
576 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
589 grad_shape = [4, 4]
590 grad_inp = np.random.rand(*grad_shape).astype("f")
592 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
609 grad_shape = [4, 4]
610 grad_inp = np.random.rand(*grad_shape).astype("f")
612 [float(x) for x in grad_inp.flatten()], shape=grad_shape)
624 grad_shape = [4, 4]
[all …]
|
| D | sparse_conditional_accumulator_test.py | 531 q.apply_grad(grad_indices=0, grad_values=[1.0], grad_shape=[]).run()
539 q.apply_grad(grad_indices=[0], grad_values=1.0, grad_shape=[]).run()
546 q.apply_grad(grad_indices=[0], grad_values=[1.0], grad_shape=[]).run()
562 grad_shape=[2, 2]).run()
571 grad_shape=[2, 3, 2]).run()
681 grad_shape=constant_op.constant(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 439 grad_shape = array_ops.shape(grad)
441 batch_shape = array_ops.slice(grad_shape, [0], [grad_rank - 2])
442 matrix_shape = array_ops.slice(grad_shape, [grad_rank - 2], [2])
457 grad_shape = array_ops.shape(grad)
458 batch_shape = grad_shape[:-2]
459 matrix_shape = grad_shape[-2:]
494 grad_shape = array_ops.shape(grad)
495 batch_shape = grad_shape[:-2]
496 matrix_shape = grad_shape[-2:]
|
| D | data_flow_ops.py | 1467 grad_shape=grad.dense_shape,
1474 grad_shape=None, argument
1515 [] if grad_shape is None else grad_shape, _dtypes.int64),
1516 has_known_shape=(grad_shape is not None),
|
| D | control_flow_state.py | 749 grad_shape = array_ops.shape_internal(enter_grad, optimize=False)
750 grad_val = array_ops.zeros(grad_shape)
|
| D | math_grad.py | 1272 grad_shape = grad._shape_tuple()
1274 return (x_shape == y_shape and x_shape == grad_shape and
|
| /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/python/ops/parallel_for/ |
| D | pfor.py | 2996 grad_shape = array_ops.shape(grad)
2997 last_dim_shape = grad_shape[-1]
2998 first_dim_shape = grad_shape[0]
|