HomeSort by relevance Sort by last modified time
    Searched refs:num_spatial_dims_ (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
conv_ops.cc 180 num_spatial_dims_(num_spatial_dims),
192 int num_dims() const { return num_spatial_dims_ + 2; }
214 for (int i = 0; i < num_spatial_dims_; ++i) {
238 const int64 in_depth = filter_shape.dim_size(num_spatial_dims_);
259 std::vector<int64> window_strides(num_spatial_dims_);
260 std::vector<int64> lhs_dilation(num_spatial_dims_, 1);
261 std::vector<int64> rhs_dilation(num_spatial_dims_);
262 std::vector<std::pair<int64, int64>> padding(num_spatial_dims_);
268 dims.set_kernel_input_feature_dimension(num_spatial_dims_);
269 dims.set_kernel_output_feature_dimension(num_spatial_dims_ + 1)
294 const int num_spatial_dims_; member in class:tensorflow::__anon38736::ConvOp
447 const int num_spatial_dims_; member in class:tensorflow::__anon38736::ConvBackpropInputOp
649 const int num_spatial_dims_; member in class:tensorflow::__anon38736::ConvBackpropFilterOp
    [all...]
pooling_ops.cc 39 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) {
63 int num_dims() const { return num_spatial_dims_ + 2; }
122 const int num_spatial_dims_; member in class:tensorflow::__anon38761::PoolingOp
245 ksize_, stride_, num_spatial_dims_,
278 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) {
286 int num_dims() const { return num_spatial_dims_ + 2; }
355 const int num_spatial_dims_; member in class:tensorflow::__anon38761::MaxPoolGradOp
389 : XlaOpKernel(ctx), num_spatial_dims_(num_spatial_dims) {
406 int num_dims() const { return num_spatial_dims_ + 2; }
442 for (int i = 0; i < num_spatial_dims_; ++i)
500 const int num_spatial_dims_; member in class:tensorflow::__anon38761::AvgPoolGradOp
    [all...]

Completed in 81 milliseconds