| /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 | 675 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
678 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 | 76 ReduceWindow(input, init, in ReduceWindowAdd()
87 ReduceWindow(input, init, in ReduceWindowMax()
98 ReduceWindow(input, init, in ReduceWindowMin()
112 ReduceWindow(input, init_value, in XLA_TEST_P()
128 ReduceWindow(input, init, CreateScalarAddComputation(FloatType(), &builder_), in XLA_TEST_P()
340 ReduceWindow( in XLA_TEST_P()
1081 ReduceWindow(/*operand=*/parameter, in XLA_TEST_P()
|
| D | cpu_gpu_fusion_test.cc | 643 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 | 472 TEST_F(HloCostAnalysisTest, ReduceWindow) { in TEST_F() argument
476 ReduceWindow(input, ConstantR0<float>(&builder, 0), add_, {4, 5}, {4, 5}, in TEST_F()
511 ReduceWindow({input1, input2}, {init, init}, compute_tuple, {4, 5}, {4, 5}, in TEST_F()
|
| D | hlo_matchers.h | 295 HLO_MATCHER(ReduceWindow);
|
| D | algebraic_simplifier_test.cc | 1855 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
1898 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
1928 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Parameter(0), in TEST_F()
4868 GmockMatch(m::ReduceWindow(m::Parameter(0), m::Constant()))); in TEST_F()
4906 EXPECT_THAT(root, GmockMatch(m::ReduceWindow(m::Convert(m::Parameter(0)), in TEST_F()
5778 GmockMatch(m::ReduceWindow(m::Op(), m::Op().Is(zero)))); in TEST_P()
8093 m::ReduceWindow(m::Parameter(0), m::ConstantScalar(0))))); in TEST_F()
|
| D | hlo.proto | 133 // broadcasting, including Reshape, Reduce, ReduceWindow, and Reverse.
|
| D | pattern_matcher.h | 2347 XLA_VARIADIC_OP_PATTERN(ReduceWindow)
|
| D | hlo_parser_test.cc | 480 "ReduceWindow", in CreateTestCases()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 714 XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
719 XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
1334 friend XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
1339 friend XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
2290 XlaOp ReduceWindow(XlaOp operand, XlaOp init_value,
2295 XlaOp ReduceWindow(absl::Span<const XlaOp> operands,
|
| D | xla_builder.cc | 2642 XlaOp XlaBuilder::ReduceWindow(XlaOp operand, XlaOp init_value, in ReduceWindow() function in xla::XlaBuilder
2647 return ReduceWindow(absl::MakeSpan(&operand, 1), in ReduceWindow()
2652 XlaOp XlaBuilder::ReduceWindow(absl::Span<const XlaOp> operands, in ReduceWindow() function in xla::XlaBuilder
4605 XlaOp ReduceWindow(const XlaOp operand, const XlaOp init_value, in ReduceWindow() function
4609 return operand.builder()->ReduceWindow(operand, init_value, computation, in ReduceWindow()
4614 XlaOp ReduceWindow(absl::Span<const XlaOp> operands, in ReduceWindow() function
4620 return operands[0].builder()->ReduceWindow(operands, init_values, computation, in ReduceWindow()
|
| D | xla_builder_test.cc | 985 ReduceWindow(gte, init, sum, /*window_dimensions=*/{1, 2, 4}, in TEST_F()
1021 ReduceWindow(input_operands, {init, init}, sum, in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2196 ## ReduceWindow section in Operation Semantics
2199 [`XlaBuilder::ReduceWindow`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_bui…
2205 `ReduceWindow`. Similar to [`Reduce`](#reduce), the applied `computation` is
2208 <b> `ReduceWindow(operands..., init_values..., computation, window_dimensions,
2252 Below code and figure shows an example of using `ReduceWindow`. Input is a
2267 // Create a ReduceWindow computation with the max reduction computation.
2271 builder.ReduceWindow(
2696 `ReduceWindow` on the `operand` array to select an element from each window, and
2719 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 | 866 op::ReduceWindow( in TEST_F()
912 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
948 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
991 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
1037 op::ReduceWindow(masked, op::Constant()))); in TEST_F()
1104 op::ReduceWindow(dim1_resharded, op::Constant()))); in TEST_F()
3303 op::ReduceWindow(masked, op::Constant())); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 482 // zeros. ReduceWindow and SelectAndScatter pads with the reduce function's
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/lhlo/IR/ |
| D | lhlo_ops.td | 547 let summary = "ReduceWindow operator";
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/ |
| D | StablehloOps.td | 2145 let summary = "ReduceWindow operator";
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 2362 let summary = "ReduceWindow operator";
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 20316 let summary = "Wraps the XLA ReduceWindow operator, documented at";
|