Searched refs:mutable_dimensions (Results 1 – 17 of 17) sorted by relevance
107 conv_window.mutable_dimensions(1)->set_size(2); in TEST_F()108 conv_window.mutable_dimensions(1)->set_window_dilation(2); in TEST_F()148 conv_window.mutable_dimensions(1)->set_size(3); in TEST_F()180 conv_window.mutable_dimensions(i)->set_size(35); in TEST_F()181 conv_window.mutable_dimensions(i)->set_padding_low(1); in TEST_F()182 conv_window.mutable_dimensions(i)->set_padding_high(1); in TEST_F()210 conv_window.mutable_dimensions(i)->set_size(4); in TEST_F()211 conv_window.mutable_dimensions(i)->set_padding_high(-1); in TEST_F()212 conv_window.mutable_dimensions(i)->set_window_dilation(2); in TEST_F()239 conv_window.mutable_dimensions(i)->set_size(35); in TEST_F()[all …]
67 padding_config.mutable_dimensions(dim)->set_edge_padding_low( in MaybePaddedAndSlicedInput()69 conv_window->mutable_dimensions(i)->set_padding_low(0); in MaybePaddedAndSlicedInput()72 padding_config.mutable_dimensions(dim)->set_edge_padding_high( in MaybePaddedAndSlicedInput()74 conv_window->mutable_dimensions(i)->set_padding_high(0); in MaybePaddedAndSlicedInput()77 padding_config.mutable_dimensions(dim)->set_interior_padding( in MaybePaddedAndSlicedInput()79 conv_window->mutable_dimensions(i)->set_base_dilation(1); in MaybePaddedAndSlicedInput()104 conv_window->mutable_dimensions(i)->set_padding_low(0); in MaybePaddedAndSlicedInput()108 conv_window->mutable_dimensions(i)->set_padding_high(0); in MaybePaddedAndSlicedInput()137 padding_config.mutable_dimensions(dim)->set_interior_padding( in MaybePaddedKernel()168 WindowDimension* dim = new_conv_window.mutable_dimensions(i); in CanonicalizeForwardConvolution()[all …]
61 pad_config.mutable_dimensions(dim)->set_edge_padding_high( in PadInstruction()
342 auto dim = new_window.mutable_dimensions(i); in MatchBackwardInput()
329 padding_config.mutable_dimensions(dim)->set_edge_padding_low( in XLA_TEST_P()331 padding_config.mutable_dimensions(dim)->set_edge_padding_high(58 + in XLA_TEST_P()350 padding_config.mutable_dimensions(0)->set_edge_padding_low(7); in XLA_TEST_P()351 padding_config.mutable_dimensions(0)->set_edge_padding_high(5); in XLA_TEST_P()352 padding_config.mutable_dimensions(0)->set_interior_padding(3); in XLA_TEST_P()353 padding_config.mutable_dimensions(1)->set_edge_padding_low(6); in XLA_TEST_P()354 padding_config.mutable_dimensions(1)->set_edge_padding_high(4); in XLA_TEST_P()355 padding_config.mutable_dimensions(1)->set_interior_padding(2); in XLA_TEST_P()375 padding_config.mutable_dimensions(dim)->set_edge_padding_low(low_padding); in XLA_TEST_P()376 padding_config.mutable_dimensions(dim)->set_edge_padding_high( in XLA_TEST_P()[all …]
57 (*result.mutable_tile_shape()->mutable_dimensions())[0]; in Tile1D()
101 auto padding_dimension = padding_config.mutable_dimensions(dim); in MakeSpatialPaddingConfig()261 auto* padding = padding_config.mutable_dimensions(dim); in AvgPoolGrad()
358 w.mutable_dimensions(0)->set_window_dilation(-1); in TEST_F()369 w.mutable_dimensions(0)->set_base_dilation(-1); in TEST_F()
97 config.mutable_dimensions(ndims - 1)->set_edge_padding_high(padding); in DiagonalBlocks()98 config.mutable_dimensions(ndims - 2)->set_edge_padding_high(padding); in DiagonalBlocks()
3074 window->mutable_dimensions(i)->set_padding_low(pad[i][0]); in ParseWindow()3075 window->mutable_dimensions(i)->set_padding_high(pad[i][1]); in ParseWindow()3078 window->mutable_dimensions(i)->set_stride(stride.empty() ? 1 : stride[i]); in ParseWindow()3079 window->mutable_dimensions(i)->set_base_dilation( in ParseWindow()3081 window->mutable_dimensions(i)->set_window_dilation( in ParseWindow()3083 window->mutable_dimensions(i)->set_window_reversal( in ParseWindow()
2210 padding_config.mutable_dimensions(i)->set_interior_padding(0); in HandlePad()2254 nonzero_padding.mutable_dimensions(i); in HandlePad()3292 auto& window_dim = *new_window.mutable_dimensions(i); in HandleReduceWindow()3427 auto& w = *new_window.mutable_dimensions(dim); in FoldConvInputPad()3493 auto& w = *new_window.mutable_dimensions(dim); in FoldConvFilterPad()
3583 padding.mutable_dimensions(1)->set_edge_padding_low(1); in TEST_F()3584 padding.mutable_dimensions(3)->set_edge_padding_high(2); in TEST_F()3665 padding.mutable_dimensions(1)->set_edge_padding_low(1); in TEST_F()3666 padding.mutable_dimensions(3)->set_edge_padding_high(2); in TEST_F()
61 proto.mutable_dimensions()->Reserve(dimensions_size()); in ToProto()
108 absl::Span<int64> mutable_dimensions() { return absl::MakeSpan(dimensions_); } in mutable_dimensions() function
242 tensor_.mutable_dimensions()->Resize(ndims + 2, 0); in BatchDescriptor()384 tensor_.mutable_dimensions()->Resize(ndims + 2, 0); in FilterDescriptor()
352 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);459 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);
439 padding.mutable_dimensions(1 + i)->set_edge_padding_high(in_size[i] - 1); in ResizeUsingDilationAndConvolutionGradOp()