Home
last modified time | relevance | path

Searched refs:SelectAndScatterWithGeneralPadding (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dselect_and_scatter_test.cc471 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()
498 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()
525 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/xla_extension/
H A Dops.pyi344 def SelectAndScatterWithGeneralPadding(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dops.cc370 &SelectAndScatterWithGeneralPadding, py::arg("operand"), in BuildOpsSubmodule()
H A Dxla_client_test.py1882 ops.SelectAndScatterWithGeneralPadding(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.h795 XlaOp SelectAndScatterWithGeneralPadding(
1401 friend XlaOp SelectAndScatterWithGeneralPadding(
2407 XlaOp SelectAndScatterWithGeneralPadding(
H A Dxla_builder.cc3295 return SelectAndScatterWithGeneralPadding( in SelectAndScatter()
3331 XlaOp XlaBuilder::SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding() function in xla::XlaBuilder
4732 XlaOp SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding() function
4738 return operand.builder()->SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dmlir_hlo_to_hlo.cc1521 value_map[op] = xla::SelectAndScatterWithGeneralPadding( in ExportXlaOp()