Home
last modified time | relevance | path

Searched refs:add_dimensions (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc329 tile->add_dimensions(512); in TEST_F()
330 tile->add_dimensions(1024); in TEST_F()
337 tile->add_dimensions(512); in TEST_F()
344 tile->add_dimensions(16); in TEST_F()
345 tile->add_dimensions(256); in TEST_F()
347 tile->add_dimensions(2); in TEST_F()
348 tile->add_dimensions(1); in TEST_F()
355 tile->add_dimensions(8); in TEST_F()
356 tile->add_dimensions(128); in TEST_F()
363 tile->add_dimensions(8); in TEST_F()
[all …]
Dshape.cc28 add_dimensions(dimension); in Shape()
63 proto.add_dimensions(dimension); in ToProto()
Dwindow_util.cc32 auto* dimension = window.add_dimensions(); in MakeWindow()
44 auto* dimension = config.add_dimensions(); in MakeSymmetricPadding()
Dutil.cc121 auto dimension = padding_config.add_dimensions(); in MakeNoPaddingConfig()
133 auto dimension = padding_config.add_dimensions(); in MakeEdgePaddingConfig()
Dlayout.cc27 tile_proto.add_dimensions(i); in ToProto()
Dlayout.h58 Tile& add_dimensions(int64 value) { in add_dimensions() function
Dshape.h99 void add_dimensions(int64 value) { in add_dimensions() function
Dshape_util.cc231 shape->add_dimensions(dimension); in PopulateShape()
271 shape->add_dimensions(bound); in AppendMajorDimension()
935 new_shape.add_dimensions(dim); in PermuteDimensions()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dpad_test.cc50 auto dimension0 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest()
54 auto dimension1 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest()
58 auto dimension2 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest()
62 auto dimension3 = r4_padding_on_dim0_dim1_.add_dimensions(); in PadTest()
91 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
106 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
121 auto dimension = padding_config.add_dimensions(); in XLA_TEST_P()
189 auto dimension0 = padding_config.add_dimensions(); in TEST_P()
193 auto dimension1 = padding_config.add_dimensions(); in TEST_P()
197 auto dimension2 = padding_config.add_dimensions(); in TEST_P()
[all …]
Dbad_rng_shape_validation_test.cc56 sans_layout.add_dimensions(1); in TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dspacetobatch_op.cc56 padding_config.add_dimensions(); // Don't pad the batch dimension. in SpaceToBatch()
58 auto* dim = padding_config.add_dimensions(); in SpaceToBatch()
70 padding_config.add_dimensions(); in SpaceToBatch()
Dpad_op.cc60 auto* dim = config.add_dimensions(); in Compile()
Dxla_pad_op.cc82 auto* dim = padding_config.add_dimensions(); in Compile()
Dstrided_slice_op.cc182 auto* dims = padding_config.add_dimensions(); in Compile()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc563 auto dimension = padding_config.add_dimensions(); in CreatePaddingConfig()
912 *window.add_dimensions() = dim; in TEST_P()
975 *window.add_dimensions() = dim; in TEST_P()
976 *window.add_dimensions() = dim; in TEST_P()
1047 *window.add_dimensions() = dim; in TEST_P()
1048 *window.add_dimensions() = dim; in TEST_P()
1125 *window.add_dimensions() = dim; in TEST_P()
1126 *window.add_dimensions() = dim; in TEST_P()
1197 *window.add_dimensions() = dim; in TEST_P()
1198 *window.add_dimensions() = dim; in TEST_P()
[all …]
Dtranspose_folding_test.cc233 WindowDimension* dim = window.add_dimensions(); in TEST_F()
289 WindowDimension* dim = window.add_dimensions(); in TEST_F()
351 WindowDimension* dim = window.add_dimensions(); in TEST_F()
418 WindowDimension* dim = window.add_dimensions(); in TEST_F()
Dshape_inference_test.cc86 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
87 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
312 *window.add_dimensions() = dim; in TEST_F()
313 *window.add_dimensions() = dim; in TEST_F()
412 auto dim0 = window.add_dimensions(); in TEST_F()
413 auto dim1 = window.add_dimensions(); in TEST_F()
457 auto dim0 = window.add_dimensions(); in TEST_F()
465 auto dim1 = window.add_dimensions(); in TEST_F()
503 auto dim0 = window.add_dimensions(); in TEST_F()
511 auto dim1 = window.add_dimensions(); in TEST_F()
[all …]
Ddynamic_dimension_inference_test.cc558 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F()
568 WindowDimension* dim = window.add_dimensions(); in TEST_F()
608 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F()
618 WindowDimension* dim = window.add_dimensions(); in TEST_F()
Dhlo_verifier_test.cc305 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
329 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
Dconvolution_group_converter.cc287 auto* dim = window.add_dimensions(); in HandleBatchGroupCount()
436 auto* dim = new_window.add_dimensions(); in HandleConvolution()
Dhlo_instructions.cc662 proto.add_dimensions(dimension); in ToProto()
700 proto.add_dimensions(dimension); in ToProto()
742 proto.add_dimensions(dimension); in ToProto()
787 proto.add_dimensions(dimension); in ToProto()
842 proto.add_dimensions(dimension); in ToProto()
880 proto.add_dimensions(dimension); in ToProto()
924 proto.add_dimensions(dimension); in ToProto()
2486 proto.add_dimensions(iota_dimension()); in ToProto()
2647 proto.add_dimensions(dimension()); in ToProto()
Dreshape_mover.cc219 shape->add_dimensions(i); in PerformSinkReshapeOrTranspose()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc56 auto dims = padding_config.add_dimensions(); in AvgPoolDivideByCountWithGeneralPadding()
95 padding_config.add_dimensions(); in MakeSpatialPaddingConfig()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc39 auto dim = conv_window_.add_dimensions(); in ConvCanonicalizationTest()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc426 instr.add_dimensions(dim); in InDimBroadcast()
602 instr.add_dimensions(iota_dimension); in Iota()
865 instr.add_dimensions(dimension); in ConcatInDim()
1235 auto dim = window.add_dimensions(); in MakeWindow()
1546 instr.add_dimensions(dim); in Transpose()
1561 instr.add_dimensions(dim); in Rev()
1701 instr.add_dimensions(dimension); in Sort()
1974 instr.add_dimensions(dim); in Reduce()
2501 instr.add_dimensions(dimension); in GetDimensionSize()

12