Searched refs:dimensions_to_reverse (Results 1 – 1 of 1) sorted by relevance
75 absl::InlinedVector<int64, 4> dimensions_to_reverse; in Compile() local90 dimensions_to_reverse.push_back(i); in Compile()95 if (!dimensions_to_reverse.empty()) { in Compile()96 slice = xla::Rev(slice, dimensions_to_reverse); in Compile()178 absl::InlinedVector<int64, 4> dimensions_to_reverse; in Compile() local196 dimensions_to_reverse.push_back(i); in Compile()207 if (!dimensions_to_reverse.empty()) { in Compile()208 grad = xla::Rev(grad, dimensions_to_reverse); in Compile()290 absl::InlinedVector<int64, 4> dimensions_to_reverse; in Compile() local307 dimensions_to_reverse.push_back(i); in Compile()[all …]