| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | padding.cc | 28 absl::Span<const int64> window_strides) { in ValidatePaddingValues()
44 absl::Span<const int64> window_strides, Padding padding) { in MakePadding()
|
| D | xla_builder.cc | 1374 absl::Span<const int64> window_strides, Padding padding, in Conv()
1386 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralPadding()
1398 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvWithGeneralDimensions()
1436 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvGeneral()
1449 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in ConvGeneralDilated()
1487 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in DynamicConvInstruction()
1529 absl::Span<const int64> window_strides, in DynamicConvInputGrad()
1552 XlaOp activations, XlaOp gradients, absl::Span<const int64> window_strides, in DynamicConvKernelGrad()
1578 XlaOp lhs, XlaOp rhs, absl::Span<const int64> window_strides, in DynamicConvForward()
1600 absl::Span<const int64> window_strides, in ConvGeneralDilatedInternal()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | xla_conv_op.cc | 47 std::vector<int64> window_strides; in Compile() local
|
| D | reduce_window_op.cc | 42 std::vector<int64> window_strides; in Compile() local
|
| D | xla_select_and_scatter_op.cc | 43 std::vector<int64> window_strides; in Compile() local
|
| D | scan_ops.cc | 78 std::vector<int64> window_strides(input_shape.dims(), 1); in Compile() local
|
| D | extract_image_patches_op.cc | 127 std::vector<int64> window_strides(num_spatial_dims); in Compile() local
|
| D | conv_op_helpers.cc | 235 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaForwardConvOp() local
439 std::vector<int64> window_strides(attrs.num_spatial_dims); in MakeXlaBackpropFilterConvOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | tree_reduction_rewriter.cc | 54 std::vector<int64> window_strides; in HandleReduce() local
|
| D | shape_inference_test.cc | 1110 std::vector<int64> window_strides = {1, 1, 1}; in TEST_F() local
1172 std::vector<int64> window_strides = {1, 1, 1}; in TEST_F() local
|
| D | shape_inference.cc | 2350 absl::Span<const int64> window_strides, in InferWindowFromDimensions()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 245 window_strides, argument
352 window_strides=None, argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_window_test.cc | 74 absl::Span<const int64> window_strides, in ReduceWindowAdd()
85 absl::Span<const int64> window_strides, in ReduceWindowMax()
96 absl::Span<const int64> window_strides, in ReduceWindowMin()
|
| D | select_and_scatter_test.cc | 46 std::vector<int64> window_strides; member
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 111 absl::Span<const int64> window_strides, in AvgPoolDivideByCount()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 390 window_strides): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 450 llvm::SmallVector<int64_t, 4> window_strides, window_dimensions; in ImportInstructionImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 100 auto window_strides = op.window_strides().getValue(); in MapWindowIvsToInput() local
|
| D | legalize_to_linalg.cc | 329 if (auto window_strides = op.window_strides()) { in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 95 absl::Span<const int64> window_strides, in ConvGeneralDilatedInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 799 llvm::SmallVector<int64_t, 5> window_strides; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 1194 SmallVector<int64_t, num_spatial_dims> window_strides; in matchAndRewrite() local
2140 ArrayAttr window_strides, StringRef padding, Builder *builder) { in GetReduceWindowPaddingAsArray()
2167 ArrayAttr window_strides, StringRef padding, Builder *builder) { in GetReduceWindowPaddingAsAttr()
4365 SmallVector<int64_t, num_spatial_dims> window_strides; in matchAndRewrite() local
5406 SmallVector<int64_t, 4> window_strides(rank, 1); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1157 mlir::DenseIntElementsAttr window_strides = op.window_strides().getValue(); in EmitConvolutionThunkFromMlir() local
|