Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc263 tile->add_dimensions(512); in TEST_F()
264 tile->add_dimensions(1024); in TEST_F()
271 tile->add_dimensions(512); in TEST_F()
278 tile->add_dimensions(16); in TEST_F()
279 tile->add_dimensions(256); in TEST_F()
281 tile->add_dimensions(2); in TEST_F()
282 tile->add_dimensions(1); in TEST_F()
289 tile->add_dimensions(8); in TEST_F()
290 tile->add_dimensions(128); in TEST_F()
297 tile->add_dimensions(8); in TEST_F()
[all …]
Dwindow_util.cc32 auto* dimension = window.add_dimensions(); in MakeWindow()
46 auto* dimension = window.add_dimensions(); in MakeWindow()
58 auto* dimension = config.add_dimensions(); in MakeSymmetricPadding()
Dshape.cc28 add_dimensions(dimension); in Shape()
63 proto.add_dimensions(dimension); in ToProto()
Dutil.cc172 auto dimension = padding_config.add_dimensions(); in MakeNoPaddingConfig()
184 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.h104 void add_dimensions(int64 value) { in add_dimensions() function
/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()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc174 *window.add_dimensions() = dim; in ReduceWindowMaxIotaTest()
175 *window.add_dimensions() = dim; in ReduceWindowMaxIotaTest()
639 auto dimension = padding_config.add_dimensions(); in CreatePaddingConfig()
988 *window.add_dimensions() = dim; in TEST_P()
1051 *window.add_dimensions() = dim; in TEST_P()
1052 *window.add_dimensions() = dim; in TEST_P()
1123 *window.add_dimensions() = dim; in TEST_P()
1124 *window.add_dimensions() = dim; in TEST_P()
1201 *window.add_dimensions() = dim; in TEST_P()
1202 *window.add_dimensions() = dim; in TEST_P()
[all …]
Dtranspose_folding_test.cc232 WindowDimension* dim = window.add_dimensions(); in TEST_F()
288 WindowDimension* dim = window.add_dimensions(); in TEST_F()
350 WindowDimension* dim = window.add_dimensions(); in TEST_F()
417 WindowDimension* dim = window.add_dimensions(); in TEST_F()
Dconvolution_group_converter.cc291 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount()
379 auto* dim = window.add_dimensions(); in HandleBatchGroupCount()
526 auto new_dim = new_window.add_dimensions(); in HandleConvolution()
641 auto window_dim = window.add_dimensions(); in HandleConvolution()
Dshape_inference_test.cc87 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
88 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
323 *window.add_dimensions() = dim; in TEST_F()
324 *window.add_dimensions() = dim; in TEST_F()
423 auto dim0 = window.add_dimensions(); in TEST_F()
424 auto dim1 = window.add_dimensions(); in TEST_F()
468 auto dim0 = window.add_dimensions(); in TEST_F()
476 auto dim1 = window.add_dimensions(); in TEST_F()
514 auto dim0 = window.add_dimensions(); in TEST_F()
522 auto dim1 = window.add_dimensions(); in TEST_F()
[all …]
Dhlo_instructions.cc628 proto.add_dimensions(*split_dimension_); in ToProto()
716 proto.add_dimensions(dimension); in ToProto()
754 proto.add_dimensions(dimension); in ToProto()
796 proto.add_dimensions(dimension); in ToProto()
841 proto.add_dimensions(dimension); in ToProto()
896 proto.add_dimensions(dimension); in ToProto()
934 proto.add_dimensions(dimension); in ToProto()
972 proto.add_dimensions(inferred_dimension_); in ToProto()
1018 proto.add_dimensions(dimension); in ToProto()
2640 proto.add_dimensions(iota_dimension()); in ToProto()
[all …]
Dconvolution_4d_expander.cc85 *new_window.add_dimensions() = instruction->window().dimensions(i); in ExpandInstruction()
Ddynamic_dimension_inference_test.cc713 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F()
723 WindowDimension* dim = window.add_dimensions(); in TEST_F()
763 WindowDimension* batch_dim = window.add_dimensions(); in TEST_F()
773 WindowDimension* dim = window.add_dimensions(); in TEST_F()
Dreshape_mover.cc219 shape->add_dimensions(i); in PerformSinkReshapeOrTranspose()
Dhlo_verifier_test.cc359 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
383 padding_config.add_dimensions()->set_interior_padding(-1); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc56 auto dims = padding_config.add_dimensions(); in AvgPoolDivideByCountWithGeneralPadding()
95 padding_config.add_dimensions(); in MakeSpatialPaddingConfig()
Dmatrix.cc192 auto* dims = padding_config.add_dimensions(); in SetMatrixDiagonal()
197 auto* dims = padding_config.add_dimensions(); in SetMatrixDiagonal()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcublas_gemm_pad_for_tensor_cores.cc70 auto dimension = padding_config.add_dimensions(); in PadForTensorCores()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization_test.cc39 auto dim = conv_window_.add_dimensions(); in ConvCanonicalizationTest()

12