Searched refs:get_dimension_size (Results 1 – 3 of 3) sorted by relevance
160 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F() local161 EXPECT_TRUE(IsConstant(get_dimension_size, &b)); in TEST_F()164 client, get_dimension_size, &b)); in TEST_F()175 auto get_dimension_size = GetDimensionSize(add, 0); in TEST_F() local177 auto add_2 = Add(get_dimension_size, get_dimension_size_2); in TEST_F()
393 HloInstruction* get_dimension_size) { in HandleGetDimensionSize() argument394 HloInstruction* operand = get_dimension_size->mutable_operand(0); in HandleGetDimensionSize()395 int64 dim = get_dimension_size->dimension(); in HandleGetDimensionSize()404 evaluated_[get_dimension_size] = in HandleGetDimensionSize()409 const Shape& shape = get_dimension_size->operand(0)->shape(); in HandleGetDimensionSize()412 static_cast<uint32>(shape.dimensions(get_dimension_size->dimension()))); in HandleGetDimensionSize()413 evaluated_[get_dimension_size] = std::move(output); in HandleGetDimensionSize()
189 Status HandleGetDimensionSize(HloInstruction* get_dimension_size) override;