Searched refs:zero_literal (Results 1 – 4 of 4) sorted by relevance
181 auto zero_literal = LiteralUtil::CreateR0(0.0f); in HandleBatchNormTraining() local182 TF_ASSIGN_OR_RETURN(zero_literal, zero_literal.Convert(ptype)); in HandleBatchNormTraining()183 auto zero = add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchNormTraining()424 auto zero_literal = LiteralUtil::CreateR0(0.0f); in HandleBatchNormGrad() local425 TF_ASSIGN_OR_RETURN(zero_literal, zero_literal.Convert(ptype)); in HandleBatchNormGrad()426 auto zero = add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchNormGrad()
365 auto zero_literal = LiteralUtil::Zero(reduce_type); in HandleBatchGroupCount() local367 add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchGroupCount()
228 Literal zero_literal = LiteralUtil::Zero(PrimitiveType::F32); in XLA_TEST_F() local231 client_->TransferToServer(zero_literal).ConsumeValueOrDie(); in XLA_TEST_F()233 XlaOp zero = Parameter(&builder, 0, zero_literal.shape(), "zero"); in XLA_TEST_F()241 Literal zero_literal = LiteralUtil::Zero(PrimitiveType::F64); in XLA_TEST_F() local244 client_->TransferToServer(zero_literal).ConsumeValueOrDie(); in XLA_TEST_F()246 XlaOp zero = Parameter(&builder, 0, zero_literal.shape(), "zero"); in XLA_TEST_F()
2170 auto zero_literal = xla::LiteralUtil::CreateR0<xla::int32>(0); in TEST() local2184 EXPECT_EQ(literal, zero_literal); in TEST()