| /external/tensorflow/tensorflow/core/util/ |
| bcast.h | 57 // grad_y = sum(grad * backprop_y(x, y), grad_y_reduce_idx)
102 const Vec& grad_y_reduce_idx() const { return grad_y_reduce_idx_; } function in class:tensorflow::BCast
|
| bcast_test.cc | 41 strings::StrAppend(&ret, "[", str_util::Join(b.grad_y_reduce_idx(), ","),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| bcast_ops.cc | 108 Output(ctx, 1, bcast.grad_y_reduce_idx());
|
| /external/tensorflow/tensorflow/core/kernels/ |
| bcast_ops.cc | 101 Output(ctx, 1, bcast.grad_y_reduce_idx());
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 624 reduce_dims[1] = bcast.grad_y_reduce_idx();
[all...] |
