OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:add_dimensions
(Results
1 - 25
of
38
) sorted by null
1
2
/external/tensorflow/tensorflow/compiler/xla/
layout_util_test.cc
329
tile->
add_dimensions
(512);
330
tile->
add_dimensions
(1024);
337
tile->
add_dimensions
(512);
344
tile->
add_dimensions
(16);
345
tile->
add_dimensions
(256);
347
tile->
add_dimensions
(2);
348
tile->
add_dimensions
(1);
355
tile->
add_dimensions
(8);
356
tile->
add_dimensions
(128);
363
tile->
add_dimensions
(8)
[
all
...]
layout.cc
27
tile_proto.
add_dimensions
(i);
shape.cc
28
add_dimensions
(dimension);
63
proto.
add_dimensions
(dimension);
window_util.cc
32
auto* dimension = window.
add_dimensions
();
44
auto* dimension = config.
add_dimensions
();
layout.h
58
Tile&
add_dimensions
(int64 value) {
function in class:xla::Tile
util.cc
121
auto dimension = padding_config.
add_dimensions
();
133
auto dimension = padding_config.
add_dimensions
();
shape.h
99
void
add_dimensions
(int64 value) {
function in class:xla::Shape
shape_util.cc
231
shape->
add_dimensions
(dimension);
271
shape->
add_dimensions
(bound);
935
new_shape.
add_dimensions
(dim);
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/tests/
bad_rng_shape_validation_test.cc
56
sans_layout.
add_dimensions
(1);
pad_test.cc
50
auto dimension0 = r4_padding_on_dim0_dim1_.
add_dimensions
();
54
auto dimension1 = r4_padding_on_dim0_dim1_.
add_dimensions
();
58
auto dimension2 = r4_padding_on_dim0_dim1_.
add_dimensions
();
62
auto dimension3 = r4_padding_on_dim0_dim1_.
add_dimensions
();
91
auto dimension = padding_config.
add_dimensions
();
106
auto dimension = padding_config.
add_dimensions
();
121
auto dimension = padding_config.
add_dimensions
();
189
auto dimension0 = padding_config.
add_dimensions
();
193
auto dimension1 = padding_config.
add_dimensions
();
197
auto dimension2 = padding_config.
add_dimensions
();
[
all
...]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
pad_op.cc
60
auto* dim = config.
add_dimensions
();
spacetobatch_op.cc
56
padding_config.
add_dimensions
(); // Don't pad the batch dimension.
58
auto* dim = padding_config.
add_dimensions
();
70
padding_config.
add_dimensions
();
xla_pad_op.cc
82
auto* dim = padding_config.
add_dimensions
();
strided_slice_op.cc
182
auto* dims = padding_config.
add_dimensions
();
/external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc
563
auto dimension = padding_config.
add_dimensions
();
[
all
...]
transpose_folding_test.cc
233
WindowDimension* dim = window.
add_dimensions
();
289
WindowDimension* dim = window.
add_dimensions
();
351
WindowDimension* dim = window.
add_dimensions
();
418
WindowDimension* dim = window.
add_dimensions
();
hlo_verifier_test.cc
305
padding_config.
add_dimensions
()->set_interior_padding(-1);
329
padding_config.
add_dimensions
()->set_interior_padding(-1);
dynamic_dimension_inference_test.cc
558
WindowDimension* batch_dim = window.
add_dimensions
();
568
WindowDimension* dim = window.
add_dimensions
();
608
WindowDimension* batch_dim = window.
add_dimensions
();
618
WindowDimension* dim = window.
add_dimensions
();
shape_inference_test.cc
86
*window_.
add_dimensions
() = dim;
87
*window_.
add_dimensions
() = dim;
312
*window.
add_dimensions
() = dim;
313
*window.
add_dimensions
() = dim;
412
auto dim0 = window.
add_dimensions
();
413
auto dim1 = window.
add_dimensions
();
457
auto dim0 = window.
add_dimensions
();
465
auto dim1 = window.
add_dimensions
();
503
auto dim0 = window.
add_dimensions
();
511
auto dim1 = window.
add_dimensions
();
[
all
...]
convolution_group_converter.cc
287
auto* dim = window.
add_dimensions
();
436
auto* dim = new_window.
add_dimensions
();
reshape_mover.cc
219
shape->
add_dimensions
(i);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
conv_canonicalization_test.cc
39
auto dim = conv_window_.
add_dimensions
();
/external/tensorflow/tensorflow/compiler/xla/client/lib/
pooling.cc
56
auto dims = padding_config.
add_dimensions
();
95
padding_config.
add_dimensions
();
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_conv_padding_legalization.cc
133
padding_config.
add_dimensions
();
cudnn_conv_rewriter.cc
164
WindowDimension* dim = backward_conv_window.
add_dimensions
();
Completed in 627 milliseconds
1
2