Searched refs:CreateScalarAndComputation (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reduce_test.cc | 132 reduce = CreateScalarAndComputation(PRED, &builder); in RunR1ToR0PredTest()
170 reduce_op = CreateScalarAndComputation(PRED, &builder); in RunR2ToR1PredTest()
803 return CreateScalarAndComputation(PRED, builder); in XLA_TEST_F()
971 auto reducer = CreateScalarAndComputation(U64, &builder); in XLA_TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.h | 55 XlaComputation CreateScalarAndComputation(PrimitiveType type,
|
| H A D | arithmetic.cc | 79 XlaComputation CreateScalarAndComputation(PrimitiveType type, in CreateScalarAndComputation() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | resampler_ops.cc | 223 xla::CreateScalarAndComputation(xla::PrimitiveType::PRED, ctx->builder()), in BoundSamples()
574 xla::CreateScalarAndComputation(xla::PrimitiveType::PRED, ctx->builder()), in Compile()
|