Searched refs:SelectAndScatterWithGeneralPadding (Results 1 – 7 of 7) sorted by relevance
471 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()498 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()525 SelectAndScatterWithGeneralPadding( in XLA_TEST_F()
344 def SelectAndScatterWithGeneralPadding(
370 &SelectAndScatterWithGeneralPadding, py::arg("operand"), in BuildOpsSubmodule()
1882 ops.SelectAndScatterWithGeneralPadding(
795 XlaOp SelectAndScatterWithGeneralPadding(1401 friend XlaOp SelectAndScatterWithGeneralPadding(2407 XlaOp SelectAndScatterWithGeneralPadding(
3295 return SelectAndScatterWithGeneralPadding( in SelectAndScatter()3331 XlaOp XlaBuilder::SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding() function in xla::XlaBuilder4732 XlaOp SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding() function4738 return operand.builder()->SelectAndScatterWithGeneralPadding( in SelectAndScatterWithGeneralPadding()
1521 value_map[op] = xla::SelectAndScatterWithGeneralPadding( in ExportXlaOp()