Searched defs:rhs_dims (Results 1 – 9 of 9) sorted by relevance
85 const auto rhs_dims = BCast::FromShape(dense_t->shape()); in Compute() local
184 auto rhs_dims = TFShapeToMklDnnDims(rhs_shape); in CreateMatMulParams() local
254 const int64 rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3()}; in Concat3D() local289 const int64 rhs_dims[] = {rhs.n1(), rhs.n2(), rhs.n3(), rhs.n4()}; in Concat4D() local
389 def window_padding_type_to_pad_values(padding_type, lhs_dims, rhs_dims, argument
1670 std::vector<int64> rhs_dims; in PartitionDotGroupOnBatch() local2103 std::vector<int64> rhs_dims; in PartitionDotGroupOnContracting() local
1249 std::array<int64_t, 1> rhs_dims{static_cast<int64_t>(op.values().size())}; in GetHoistParams() local
2184 int64 rhs_dims = hlo->operand(1)->shape().dimensions_size(); in EmitElementalDot() local
3715 std::vector<string> rhs_dims(2 + dnums.kernel_spatial_dimensions().size()); in ConvolutionDimensionNumbersToString() local
5178 std::vector<int64> rhs_dims = lhs_dims; in TEST_P() local