Home
last modified time | relevance | path

Searched refs:grad_y_reduce_idx (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbcast_ops.cc108 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dbcast_ops.cc101 Output(ctx, 1, bcast.grad_y_reduce_idx()); in Compute()
/external/tensorflow/tensorflow/core/util/
Dbcast_test.cc40 strings::StrAppend(&ret, "[", absl::StrJoin(b.grad_y_reduce_idx(), ","), "]"); in BCast()
Dbcast.h381 const Vec& grad_y_reduce_idx() const { return grad_reduce_idx_[1]; } in grad_y_reduce_idx() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc678 reduce_dims[1] = bcast.grad_y_reduce_idx(); in MaterializeBroadcastGradientArgs()