HomeSort by relevance Sort by last modified time
    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