Searched refs:window_dims (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scan_ops.cc | 79 std::vector<int64> window_dims(input_shape.dims(), 1); in Compile() local
80 window_dims[axis] = input_shape.dim_size(axis); in Compile()
104 window_dims, window_strides, in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | pooling.cc | 38 std::vector<int64> window_dims(num_spatial_dims); in AvgPoolDivideByCountWithGeneralPadding() local
46 window_dims[i] = dim; in AvgPoolDivideByCountWithGeneralPadding()
69 return Div(sums, counts, window_dims); in AvgPoolDivideByCountWithGeneralPadding()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 5286 const auto& window_dims = convolution->window().dimensions(); in SwapConvOperands() local
5296 const int64 kernel_size = window_dims[spatial_dim].size(); in SwapConvOperands()
5298 !window_dims[spatial_dim].window_reversal() && in SwapConvOperands()
5299 window_dims[spatial_dim].padding_low() == 0 && in SwapConvOperands()
5300 window_dims[spatial_dim].padding_high() == 0 && in SwapConvOperands()
5301 window_dims[spatial_dim].window_dilation() == 1; in SwapConvOperands()
5304 window_dims[spatial_dim].base_dilation() == kernel_size && in SwapConvOperands()
5305 window_dims[spatial_dim].stride() == kernel_size - 1; in SwapConvOperands()
5310 window_dims[spatial_dim].base_dilation() == 1 && in SwapConvOperands()
5311 window_dims[spatial_dim].stride() == 1; in SwapConvOperands()
[all …]
|
| D | shape_inference.cc | 1799 std::vector<int64> window_dims(num_spatial_dims); in InferConvolveShape() local
1801 window_dims[i] = window.dimensions(i).size(); in InferConvolveShape()
1803 if (kernel_spatial_dims != window_dims) { in InferConvolveShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2139 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsArray() argument
2152 for (Attribute attr : window_dims) in GetReduceWindowPaddingAsArray()
2166 llvm::ArrayRef<int64_t> input_dims, ArrayAttr window_dims, in GetReduceWindowPaddingAsAttr() argument
2171 input_dims, window_dims, window_strides, padding, builder); in GetReduceWindowPaddingAsAttr()
3938 int64_t window_dims = tensor_rank - num_index_dims; in matchAndRewrite() local
3940 GetI64ElementsAttrForSeq(updates_rank - window_dims, updates_rank, in matchAndRewrite()
5405 SmallVector<int64_t, 4> window_dims(rank, 1); in matchAndRewrite() local
5407 window_dims[axis] = input_shape[axis]; in matchAndRewrite()
5422 GetI64ElementsAttr(rewriter.getI64ArrayAttr(window_dims)), in matchAndRewrite()
|