Searched refs:SelectAndScatter (Results 1 - 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | select_and_scatter_test.cc | 83 auto select_and_scatter = builder_.SelectAndScatter(
203 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3},
215 builder_.SelectAndScatter(operand, ge_f32_, /*window_dimensions=*/{3},
226 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3},
237 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{3},
249 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3},
263 builder_.SelectAndScatter(reshape, ge_s32_, /*window_dimensions=*/{2, 3},
275 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 3},
287 builder_.SelectAndScatter(operand, ge_s32_, /*window_dimensions=*/{2, 2},
301 builder_.SelectAndScatter(operan
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_cost_analysis_test.cc | 248 TEST_F(HloCostAnalysisTest, SelectAndScatter) {
255 builder.SelectAndScatter(operand, gt_, {4, 5}, {4, 5}, Padding::kValid,
|
| H A D | hlo_matchers.h | 152 HLO_MATCHER(SelectAndScatter);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | pooling_ops.cc | 347 xla::ComputationDataHandle gradients = ctx->builder()->SelectAndScatter(
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | xla_client_test.py | 1042 c.SelectAndScatter(c.Constant(NumpyArrayF32([[1., 2., 6.], [4., 5., 3.]])),
1054 c.SelectAndScatter(c.Constant(NumpyArrayF64([[1., 2., 6.], [4., 5., 3.]])),
|
| H A D | xla_client.py | 713 def SelectAndScatter(self, operand, select, window_dimensions, window_strides, member in class:ComputationBuilder
731 A ComputationDataHandle representing the added SelectAndScatter op.
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | computation_builder.h | 563 ComputationDataHandle SelectAndScatter(
570 // As SelectAndScatter(), but the padding is given in the format
|
| H A D | computation_builder.cc | 1409 ComputationDataHandle ComputationBuilder::SelectAndScatter( function in class:xla::ComputationBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| H A D | hlo_parser_test.cc | 411 "SelectAndScatter",
|
Completed in 153 milliseconds