Searched refs:a_prime (Results 1 – 2 of 2) sorted by relevance
129 Tensor a_prime(DT_FLOAT); in TEST_F() local130 ASSERT_TRUE(a_prime.FromProto(event.summary().value(0).tensor())); in TEST_F()133 ASSERT_EQ(tensor_a_->shape(), a_prime.shape()); in TEST_F()134 for (int i = 0; i < a_prime.flat<float>().size(); ++i) { in TEST_F()135 ASSERT_EQ(tensor_a_->flat<float>()(i), a_prime.flat<float>()(i)); in TEST_F()293 Tensor a_prime(DT_FLOAT); in TEST_F() local294 ASSERT_TRUE(a_prime.FromProto(event.summary().value(0).tensor())); in TEST_F()297 ASSERT_EQ(tensor_a_->shape(), a_prime.shape()); in TEST_F()298 for (int i = 0; i < a_prime.flat<float>().size(); ++i) { in TEST_F()299 ASSERT_EQ(tensor_a_->flat<float>()(i), a_prime.flat<float>()(i)); in TEST_F()[all …]
682 Value a_prime = zero; in MaterializeDigamma() local687 a_prime = rewriter.create<mhlo::SubOp>( in MaterializeDigamma()688 loc, a_prime, in MaterializeDigamma()712 Value a_prime_div_a = rewriter.create<mhlo::DivOp>(loc, a_prime, a); in MaterializeDigamma()