Searched defs:expected_dims (Results 1 – 6 of 6) sorted by relevance
28 const gtl::ArraySlice<int64> expected_dims) { in TestDimensionReduction()
365 def testBroadcastDynamicShape(self, x_dims, y_dims, expected_dims): argument
143 auto expected_dims = str_util::Split(expected, ','); in InferShapes() local
403 absl::InlinedVector<int64_t, 4> expected_dims; member
487 size_t expected_dims = expected_shape.dims(); in Compute() local
286 const std::vector<int>& expected_dims, in ValidateWeights()1944 const std::vector<OutputCType>& expected_value) { in TestConvertConst()1981 std::vector<int> expected_dims{1}; in TestConvertConst() local