Searched refs:is_backprop (Results 1 – 1 of 1) sorted by relevance
120 int64 num_splits, bool is_backprop = false, bool is_rhs = false);164 bool is_backprop = false, bool is_rhs = false);481 bool is_backprop, bool is_rhs) { in BringSpaceNextToBatch() argument491 CHECK(is_backprop); in BringSpaceNextToBatch()535 if (is_backprop && i == dim_numbers.input_batch_dimension()) { in BringSpaceNextToBatch()555 if (is_backprop) { in BringSpaceNextToBatch()2096 int64 num_splits, bool is_backprop, in SplitSpace() argument2101 activations_batch_dim, is_backprop, is_rhs)); in SplitSpace()