Home
last modified time | relevance | path

Searched refs:CreateScalarGeComputation (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dselect_and_scatter_test.cc57 ge_s32_ = CreateScalarGeComputation(S32, &builder_); in SelectAndScatterTest()
59 ge_f32_ = CreateScalarGeComputation(F32, &builder_); in SelectAndScatterTest()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Darithmetic.h43 XlaComputation CreateScalarGeComputation(PrimitiveType type,
H A Darithmetic.cc61 XlaComputation CreateScalarGeComputation(PrimitiveType type, in CreateScalarGeComputation() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dpooling_ops.cc404 auto select = CreateScalarGeComputation(element_type, ctx->builder()); in Compile()