Searched refs:rhs_batch (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | tridiagonal_matmul_op_test.py | 61 rhs_batch = array_ops.stack([rhs, 2 * rhs])
76 diags_sequence_batch, rhs_batch, diagonals_format='sequence'),
78 diags_compact_batch, rhs_batch, diagonals_format='compact'),
80 diags_matrix_batch, rhs_batch, diagonals_format='matrix')
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 519 (lhs_contract, rhs_contract), (lhs_batch, rhs_batch) = dimension_numbers
524 dot_dims_proto.rhs_batch_dimensions.extend(rhs_batch)
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 822 const int64 rhs_batch = in CanPropagate() local
845 if (rhs_batch * kNumSplits != lhs_batch) { in CanPropagate()
867 const int64 rhs_batch = second_operand->shape().dimensions(new_batch_dim); in CanPropagate() local
873 if (rhs_batch != kNumSplits * lhs_batch) { in CanPropagate()
|