| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lrn_ops.cc | 54 auto reduce = xla::ReduceWindow( in Compile()
141 auto reduce = xla::ReduceWindow( in Compile()
158 auto dy_reduce = xla::ReduceWindow( in Compile()
|
| D | pooling_ops.cc | 600 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
603 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_lo, xla::F32), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 66 ReduceWindow(padded_ones, zero, CreateScalarAddComputation(dtype, b), in AvgPoolDivideByCountWithGeneralPadding()
83 return ReduceWindow(operand, init_value, add_computation, kernel_size, in ComputeSums()
142 return ReduceWindow(operand, init_value, max_computation, kernel_size, in MaxPool()
273 ReduceWindow(padded_gradients, Zero(b, dtype), in AvgPoolGrad()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 78 ReduceWindow(input, init, in ReduceWindowAdd()
89 ReduceWindow(input, init, in ReduceWindowMax()
100 ReduceWindow(input, init, in ReduceWindowMin()
114 ReduceWindow(input, init_value, in TEST_P()
130 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in TEST_P()
319 ReduceWindow( in XLA_TEST_P()
1040 ReduceWindow(/*operand=*/parameter, in TEST_P()
|
| D | fusion_test.cc | 647 XLA_TEST_F(FusionTest, DISABLED_ON_CPU(ReduceWindow)) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1554 c.ReduceWindow(operand=c.Constant(input_array),
1564 c.ReduceWindow(operand=c.Constant(input_array),
1574 c.ReduceWindow(operand=c.Constant(input_array),
1584 c.ReduceWindow(operand=c.Constant(input_array),
1594 c.ReduceWindow(operand=c.Constant(input_array),
1604 c.ReduceWindow(operand=c.Constant(input_array),
|
| D | xla_client.py | 1472 def ReduceWindow(self, operand, init_value, computation_to_apply, member in ComputationBuilder
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 334 TEST_F(HloCostAnalysisTest, ReduceWindow) { in TEST_F() argument
338 ReduceWindow(input, ConstantR0<float>(&builder, 0), add_, {4, 5}, {4, 5}, in TEST_F()
|
| D | hlo_matchers.h | 227 HLO_MATCHER(ReduceWindow);
|
| D | hlo.proto | 72 // broadcasting, including Reshape, Reduce, ReduceWindow, and Reverse.
|
| D | algebraic_simplifier_test.cc | 1421 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
1464 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
3637 GmockMatch(m::ReduceWindow(m::Op().Is(operand), m::Constant()))); in TEST_F()
3722 EXPECT_THAT(root, GmockMatch(m::ReduceWindow(m::Convert(m::Parameter(0)), in TEST_F()
4280 GmockMatch(m::ReduceWindow(m::Op(), m::Op().Is(zero)))); in TEST_P()
|
| D | hlo_parser_test.cc | 368 "ReduceWindow", in CreateTestCases()
|
| D | pattern_matcher.h | 2174 XLA_BINOP_PATTERN(ReduceWindow)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 452 XlaOp ReduceWindow(const XlaOp& operand, const XlaOp& init_value,
871 friend XlaOp ReduceWindow(const XlaOp& operand, const XlaOp& init_value,
1548 XlaOp ReduceWindow(const XlaOp& operand, const XlaOp& init_value,
|
| D | xla_builder_test.cc | 789 ReduceWindow(gte, init, sum, /*window_dimensions=*/{1, 2, 4}, in TEST_F()
|
| D | xla_builder.cc | 1993 XlaOp XlaBuilder::ReduceWindow(const XlaOp& operand, const XlaOp& init_value, in ReduceWindow() function in xla::XlaBuilder
3210 XlaOp ReduceWindow(const XlaOp& operand, const XlaOp& init_value, in ReduceWindow() function
3214 return operand.builder()->ReduceWindow(operand, init_value, computation, in ReduceWindow()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 1858 ## ReduceWindow section in Operation Semantics
1861 [`XlaBuilder::ReduceWindow`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_bui…
1866 pooling layer can be expressed as a `ReduceWindow`. Similar to
1870 <b> `ReduceWindow(operand, init_value, computation, window_dimensions,
1897 Below code and figure shows an example of using `ReduceWindow`. Input is a
1912 // Create a ReduceWindow computation with the max reduction computation.
1916 builder.ReduceWindow(
2273 `ReduceWindow` on the `operand` array to select an element from each window, and
2296 array must have the same shape as the result of applying a `ReduceWindow`
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 399 // zeros. ReduceWindow and SelectAndScatter pads with the reduce function's
|