Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.h36 XlaComputation CreateScalarGeComputation(PrimitiveType type,
Darithmetic.cc72 XlaComputation CreateScalarGeComputation(PrimitiveType type, in CreateScalarGeComputation() function
/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_and_scatter_test.cc55 ge_s32_ = CreateScalarGeComputation(S32, &builder_); in SelectAndScatterTest()
57 ge_f32_ = CreateScalarGeComputation(F32, &builder_); in SelectAndScatterTest()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dpooling_ops.cc335 auto select = CreateScalarGeComputation(element_type, ctx->builder()); in Compile()