| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 55 // grad_x = sum(grad * backprop_x(x, y), grad_x_reduce_idx)
101 const Vec& grad_x_reduce_idx() const { return grad_x_reduce_idx_; } function in class:tensorflow::BCast
|
| bcast_test.cc | 39 strings::StrAppend(&ret, "[", str_util::Join(b.grad_x_reduce_idx(), ","),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bcast_ops.cc | 107 Output(ctx, 0, bcast.grad_x_reduce_idx());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| bcast_ops.cc | 100 Output(ctx, 0, bcast.grad_x_reduce_idx());
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 623 reduce_dims[0] = bcast.grad_x_reduce_idx();
[all...] |
