Searched refs:CreateScalarAndComputation (Results 1 – 4 of 4) sorted by relevance
55 XlaComputation CreateScalarAndComputation(PrimitiveType type,
79 XlaComputation CreateScalarAndComputation(PrimitiveType type, in CreateScalarAndComputation() function
131 reduce = CreateScalarAndComputation(PRED, &builder); in RunR1ToR0PredTest()169 reduce_op = CreateScalarAndComputation(PRED, &builder); in RunR2ToR1PredTest()796 return CreateScalarAndComputation(PRED, builder); in XLA_TEST_F()964 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()