Home
last modified time | relevance | path

Searched refs:dim_in_range (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dxla_reduce_op.cc52 auto dim_in_range = [rank](int64 dim) { return dim >= 0 && dim < rank; }; in Compile() local
55 absl::c_all_of(dimensions_to_reduce_, dim_in_range), in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/ops/
Dxla_ops.cc326 auto dim_in_range = [rank](int64 dim) { in __anon3500048a0302() local
331 !absl::c_all_of(dimensions_to_reduce, dim_in_range)) { in __anon3500048a0302()