OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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