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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_rewriter_test.cc 107 conv_window.mutable_dimensions(1)->set_size(2);
108 conv_window.mutable_dimensions(1)->set_window_dilation(2);
148 conv_window.mutable_dimensions(1)->set_size(3);
180 conv_window.mutable_dimensions(i)->set_size(35);
181 conv_window.mutable_dimensions(i)->set_padding_low(1);
182 conv_window.mutable_dimensions(i)->set_padding_high(1);
210 conv_window.mutable_dimensions(i)->set_size(4);
211 conv_window.mutable_dimensions(i)->set_padding_high(-1);
212 conv_window.mutable_dimensions(i)->set_window_dilation(2);
239 conv_window.mutable_dimensions(i)->set_size(35)
    [all...]
cudnn_conv_padding_legalization.cc 67 padding_config.mutable_dimensions(dim)->set_edge_padding_low(
69 conv_window->mutable_dimensions(i)->set_padding_low(0);
72 padding_config.mutable_dimensions(dim)->set_edge_padding_high(
74 conv_window->mutable_dimensions(i)->set_padding_high(0);
77 padding_config.mutable_dimensions(dim)->set_interior_padding(
79 conv_window->mutable_dimensions(i)->set_base_dilation(1);
104 conv_window->mutable_dimensions(i)->set_padding_low(0);
108 conv_window->mutable_dimensions(i)->set_padding_high(0);
137 padding_config.mutable_dimensions(dim)->set_interior_padding(
168 WindowDimension* dim = new_conv_window.mutable_dimensions(i)
    [all...]
cudnn_conv_pad_for_tensor_cores.cc 61 pad_config.mutable_dimensions(dim)->set_edge_padding_high(
cudnn_conv_rewriter.cc 342 auto dim = new_window.mutable_dimensions(i);
  /external/tensorflow/tensorflow/compiler/xla/tests/
pad_test.cc 329 padding_config.mutable_dimensions(dim)->set_edge_padding_low(
331 padding_config.mutable_dimensions(dim)->set_edge_padding_high(58 +
350 padding_config.mutable_dimensions(0)->set_edge_padding_low(7);
351 padding_config.mutable_dimensions(0)->set_edge_padding_high(5);
352 padding_config.mutable_dimensions(0)->set_interior_padding(3);
353 padding_config.mutable_dimensions(1)->set_edge_padding_low(6);
354 padding_config.mutable_dimensions(1)->set_edge_padding_high(4);
355 padding_config.mutable_dimensions(1)->set_interior_padding(2);
375 padding_config.mutable_dimensions(dim)->set_edge_padding_low(low_padding);
376 padding_config.mutable_dimensions(dim)->set_edge_padding_high
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
sharding_builder.cc 57 (*result.mutable_tile_shape()->mutable_dimensions())[0];
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
pooling.cc 101 auto padding_dimension = padding_config.mutable_dimensions(dim);
261 auto* padding = padding_config.mutable_dimensions(dim);
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_verifier_test.cc 358 w.mutable_dimensions(0)->set_window_dilation(-1);
369 w.mutable_dimensions(0)->set_base_dilation(-1);
triangular_solve_expander.cc 97 config.mutable_dimensions(ndims - 1)->set_edge_padding_high(padding);
98 config.mutable_dimensions(ndims - 2)->set_edge_padding_high(padding);
hlo_parser.cc     [all...]
algebraic_simplifier.cc     [all...]
algebraic_simplifier_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape.cc 61 proto.mutable_dimensions()->Reserve(dimensions_size());
shape.h 108 absl::Span<int64> mutable_dimensions() { return absl::MakeSpan(dimensions_); } function in class:xla::Shape
  /external/tensorflow/tensorflow/stream_executor/
dnn.cc 242 tensor_.mutable_dimensions()->Resize(ndims + 2, 0);
384 tensor_.mutable_dimensions()->Resize(ndims + 2, 0);
dnn.h 352 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);
459 return AsInt64Slice(tensor_.mutable_dimensions()).subspan(2);
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_resize_ops.cc 439 padding.mutable_dimensions(1 + i)->set_edge_padding_high(in_size[i] - 1);

Completed in 2229 milliseconds