| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | select_and_scatter_test.cc | 83 SelectAndScatter(operand, ge_f32_, GetParam().window_dimensions, in XLA_TEST_P()
208 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
220 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
231 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
242 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3}, in XLA_TEST_F()
254 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
269 SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3, 3}, in XLA_TEST_F()
283 SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
295 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3}, in XLA_TEST_F()
307 SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 2}, in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 351 TEST_F(HloCostAnalysisTest, SelectAndScatter) { in TEST_F() argument
357 SelectAndScatter(operand, gt_, {4, 5}, {4, 5}, Padding::kValid, source, in TEST_F()
|
| D | hlo_matchers.h | 234 HLO_MATCHER(SelectAndScatter);
|
| D | hlo_parser_test.cc | 471 "SelectAndScatter", in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 484 XlaOp SelectAndScatter(const XlaOp& operand, const XlaComputation& select,
897 friend XlaOp SelectAndScatter(const XlaOp& operand,
1614 XlaOp SelectAndScatter(const XlaOp& operand, const XlaComputation& select,
|
| D | xla_builder_test.cc | 828 SelectAndScatter(gte0, ge, {1, 2, 4}, {1, 2, 4}, Padding::kValid, source, in TEST_F()
|
| D | xla_builder.cc | 2244 XlaOp XlaBuilder::SelectAndScatter(const XlaOp& operand, in SelectAndScatter() function in xla::XlaBuilder
3259 XlaOp SelectAndScatter(const XlaOp& operand, const XlaComputation& select, in SelectAndScatter() function
3264 return operand.builder()->SelectAndScatter(operand, select, window_dimensions, in SelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 338 xla::SelectAndScatter(input, select, ksize_, stride_, xla_padding, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1432 c.SelectAndScatter(c.Constant(NumpyArrayF32([[1., 2., 6.], [4., 5., 3.]])),
1444 c.SelectAndScatter(c.Constant(NumpyArrayF64([[1., 2., 6.], [4., 5., 3.]])),
|
| D | xla_client.py | 1282 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, member in ComputationBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2267 ## SelectAndScatter section in Operation Semantics
2270 [`XlaBuilder::SelectAndScatter`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla…
2297 operation on the `operand` array. `SelectAndScatter` can be used to
2300 <b>`SelectAndScatter(operand, select, window_dimensions, window_strides,
2330 The figure below shows examples of using `SelectAndScatter`, with the `select`
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 399 // zeros. ReduceWindow and SelectAndScatter pads with the reduce function's
|