Searched defs:rhs_dims (Results 1 – 6 of 6) sorted by relevance
85 const auto rhs_dims = BCast::FromShape(dense_t->shape()); in Compute() local
262 const int64 rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3()}; in Concat3D() local297 const int64 rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; in Concat4D() local
2125 int64 rhs_dims = hlo->operand(1)->shape().dimensions_size(); in EmitElementalDot() local
3061 std::vector<string> rhs_dims(2 + dnums.kernel_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
4321 std::vector<int64> rhs_dims = lhs_dims; in TEST_P() local
250 def _convert_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument