Searched defs:GetDimensionSize (Results 1 – 7 of 7) sorted by relevance
1643 ## GetDimensionSize section in Operation Semantics
154 TEST_F(ComputeConstantTest, GetDimensionSize) { in TEST_F() argument
218 TEST_F(DynamismInferenceTest, GetDimensionSize) { in TEST_F() argument525 TEST_F(UpperBoundInferenceTest, GetDimensionSize) { in TEST_F() argument
473 TEST_F(XlaBuilderTest, GetDimensionSize) { in TEST_F() argument
3415 XlaOp XlaBuilder::GetDimensionSize(XlaOp operand, int64_t dimension) { in GetDimensionSize() function in xla::XlaBuilder4832 XlaOp GetDimensionSize(const XlaOp operand, int64_t dimension) { in GetDimensionSize() function
1053 const auto GetDimensionSize = [](auto& d) { return d.size(); }; in ParseConst() local
4256 TEST_F(HloEvaluatorTest, GetDimensionSize) { in TEST_F() argument