| /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 | 623 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
626 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()
143 return ReduceWindow(operand, init_value, max_computation, kernel_size, in MaxPool()
276 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 XLA_TEST_P()
130 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
342 ReduceWindow( in XLA_TEST_P()
1056 ReduceWindow(/*operand=*/parameter, in XLA_TEST_P()
|
| D | cpu_gpu_fusion_test.cc | 645 XLA_TEST_F(CpuGpuFusionTest, DISABLED_ON_CPU(ReduceWindow)) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_cost_analysis_test.cc | 428 TEST_F(HloCostAnalysisTest, ReduceWindow) { in TEST_F() argument
432 ReduceWindow(input, ConstantR0<float>(&builder, 0), add_, {4, 5}, {4, 5}, in TEST_F()
467 ReduceWindow({input1, input2}, {init, init}, compute_tuple, {4, 5}, {4, 5}, in TEST_F()
|
| D | hlo_matchers.h | 255 HLO_MATCHER(ReduceWindow);
|
| D | hlo.proto | 78 // broadcasting, including Reshape, Reduce, ReduceWindow, and Reverse.
|
| D | algebraic_simplifier_test.cc | 1785 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
1828 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
4405 GmockMatch(m::ReduceWindow(m::Op().Is(operand), m::Constant()))); in TEST_F()
4490 EXPECT_THAT(root, GmockMatch(m::ReduceWindow(m::Convert(m::Parameter(0)), in TEST_F()
5137 GmockMatch(m::ReduceWindow(m::Op(), m::Op().Is(zero)))); in TEST_P()
|
| D | pattern_matcher.h | 2090 XLA_BINOP_PATTERN(ReduceWindow)
|
| D | hlo_parser_test.cc | 432 "ReduceWindow", in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 700 XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
705 XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
1277 friend XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
1282 friend XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
2161 XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
2166 XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
|
| D | xla_builder.cc | 2487 XlaOp XlaBuilder::ReduceWindow(XlaOp operand, XlaOp init_value, in ReduceWindow() function in xla::XlaBuilder
2492 return ReduceWindow(absl::MakeSpan(&operand, 1), in ReduceWindow()
2497 XlaOp XlaBuilder::ReduceWindow(absl::Span<const XlaOp> operands, in ReduceWindow() function in xla::XlaBuilder
4619 XlaOp ReduceWindow(const XlaOp operand, const XlaOp init_value, in ReduceWindow() function
4623 return operand.builder()->ReduceWindow(operand, init_value, computation, in ReduceWindow()
4628 XlaOp ReduceWindow(absl::Span<const XlaOp> operands, in ReduceWindow() function
4634 return operands[0].builder()->ReduceWindow(operands, init_values, computation, in ReduceWindow()
|
| D | xla_builder_test.cc | 887 ReduceWindow(gte, init, sum, /*window_dimensions=*/{1, 2, 4}, in TEST_F()
923 ReduceWindow(input_operands, {init, init}, sum, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2087 ## ReduceWindow section in Operation Semantics
2090 [`XlaBuilder::ReduceWindow`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_bui…
2095 pooling layer can be expressed as a `ReduceWindow`. Similar to
2099 <b> `ReduceWindow(operand, init_value, computation, window_dimensions,
2130 Below code and figure shows an example of using `ReduceWindow`. Input is a
2145 // Create a ReduceWindow computation with the max reduction computation.
2149 builder.ReduceWindow(
2559 `ReduceWindow` on the `operand` array to select an element from each window, and
2582 array must have the same shape as the result of applying a `ReduceWindow`
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_test.cc | 831 op::ReduceWindow( in TEST_F()
877 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
914 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
957 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
1003 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
1070 op::ReduceWindow(dim1_resharded, op::Constant()))); in TEST_F()
3078 op::ReduceWindow(masked, op::Constant())); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 434 // zeros. ReduceWindow and SelectAndScatter pads with the reduce function's
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops_base.td | 691 string summary = "ReduceWindow operator";
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 17938 let summary = "Wraps the XLA ReduceWindow operator, documented at";
|