Searched refs:zero_literal (Results 1 – 3 of 3) sorted by relevance
214 auto zero_literal = LiteralUtil::CreateR0(0.0f); in HandleBatchNormTraining() local215 TF_ASSIGN_OR_RETURN(zero_literal, zero_literal.Convert(ptype)); in HandleBatchNormTraining()216 auto zero = add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchNormTraining()475 auto zero_literal = LiteralUtil::CreateR0(0.0f); in HandleBatchNormGrad() local476 TF_ASSIGN_OR_RETURN(zero_literal, zero_literal.Convert(ptype)); in HandleBatchNormGrad()477 auto zero = add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchNormGrad()
268 auto zero_literal = LiteralUtil::Zero(reduce_type); in HandleBatchGroupCount() local270 add(HloInstruction::CreateConstant(std::move(zero_literal))); in HandleBatchGroupCount()
216 Literal zero_literal = LiteralUtil::Zero(PrimitiveType::F32); in XLA_TEST_F() local219 client_->TransferToServer(zero_literal).ConsumeValueOrDie(); in XLA_TEST_F()221 XlaOp zero = Parameter(&builder, 0, zero_literal.shape(), "zero"); in XLA_TEST_F()