Searched refs:CreateScalarAndComputation (Results 1 – 4 of 4) sorted by relevance
48 XlaComputation CreateScalarAndComputation(PrimitiveType type,
97 XlaComputation CreateScalarAndComputation(PrimitiveType type, in CreateScalarAndComputation() function
130 reduce = CreateScalarAndComputation(PRED, &builder); in RunR1ToR0PredTest()168 reduce_op = CreateScalarAndComputation(PRED, &builder); in RunR2ToR1PredTest()795 return CreateScalarAndComputation(PRED, builder); in XLA_TEST_F()963 auto reducer = CreateScalarAndComputation(U64, &builder); in XLA_TEST_F()
220 xla::CreateScalarAndComputation(xla::PrimitiveType::PRED, ctx->builder()), in BoundSamples()572 xla::CreateScalarAndComputation(xla::PrimitiveType::PRED, in Compile()