Home
last modified time | relevance | path

Searched refs:CustomCallWithComputation (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
H A Dmlir_hlo_builder_test.cc188 TEST_F(XlaBuilderTest, CustomCallWithComputation) { in TEST_F() argument
202 auto custom_call = CustomCallWithComputation( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dapprox_topk.cc237 auto approx_topk = CustomCallWithComputation( in ApproxTopK()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h1267 friend XlaOp CustomCallWithComputation(
2142 XlaOp CustomCallWithComputation(
H A Dxla_builder.cc4435 XlaOp CustomCallWithComputation( in CustomCallWithComputation() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc1120 value_map[result] = xla::CustomCallWithComputation( in ExportXlaOp()