Searched defs:GetDimensionSize (Results 1 – 6 of 6) sorted by relevance
1679 ## GetDimensionSize section in Operation Semantics
153 TEST_F(ComputeConstantTest, GetDimensionSize) { in TEST_F() argument
244 TEST_F(DynamismInferenceTest, GetDimensionSize) { in TEST_F() argument577 TEST_F(UpperBoundInferenceTest, GetDimensionSize) { in TEST_F() argument
498 TEST_F(XlaBuilderTest, GetDimensionSize) { in TEST_F() argument
3558 XlaOp XlaBuilder::GetDimensionSize(XlaOp operand, int64_t dimension) { in GetDimensionSize() function in xla::XlaBuilder5004 XlaOp GetDimensionSize(const XlaOp operand, int64_t dimension) { in GetDimensionSize() function
4354 TEST_F(HloEvaluatorTest, GetDimensionSize) { in TEST_F() argument