Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc281 tile->add_dimensions(512); in TEST_F()
282 tile->add_dimensions(1024); in TEST_F()
289 tile->add_dimensions(512); in TEST_F()
296 tile->add_dimensions(16); in TEST_F()
297 tile->add_dimensions(256); in TEST_F()
299 tile->add_dimensions(2); in TEST_F()
300 tile->add_dimensions(1); in TEST_F()
307 tile->add_dimensions(8); in TEST_F()
308 tile->add_dimensions(128); in TEST_F()
315 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()
Dlayout.cc27 tile_proto.add_dimensions(i); in ToProto()
Dutil.cc136 auto dimension = padding_config.add_dimensions(); in MakeNoPaddingConfig()
148 auto dimension = padding_config.add_dimensions(); in MakeEdgePaddingConfig()
Dlayout.h58 Tile& add_dimensions(int64 value) { in add_dimensions() function
Dshape.h115 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/xla/service/
Dall_to_all_decomposer.cc71 new_all_to_all_shape.add_dimensions(all_to_all->shape().dimensions(i)); in ExpandInstruction()
74 new_all_to_all_shape.add_dimensions(all_to_all_group_size); in ExpandInstruction()
75 new_all_to_all_shape.add_dimensions(split_size); in ExpandInstruction()
77 new_all_to_all_shape.add_dimensions(1); in ExpandInstruction()
Dhlo_evaluator_test.cc175 *window.add_dimensions() = dim; in ReduceWindowMaxIotaTest()
176 *window.add_dimensions() = dim; in ReduceWindowMaxIotaTest()
640 auto dimension = padding_config.add_dimensions(); in CreatePaddingConfig()
989 *window.add_dimensions() = dim; in TEST_P()
1052 *window.add_dimensions() = dim; in TEST_P()
1053 *window.add_dimensions() = dim; in TEST_P()
1124 *window.add_dimensions() = dim; in TEST_P()
1125 *window.add_dimensions() = dim; in TEST_P()
1202 *window.add_dimensions() = dim; in TEST_P()
1203 *window.add_dimensions() = dim; in TEST_P()
[all …]
Dtranspose_folding_test.cc234 WindowDimension* dim = window.add_dimensions(); in TEST_F()
291 WindowDimension* dim = window.add_dimensions(); in TEST_F()
354 WindowDimension* dim = window.add_dimensions(); in TEST_F()
422 WindowDimension* dim = window.add_dimensions(); in TEST_F()
Dconvolution_group_converter.cc293 auto window_dim = window.add_dimensions(); in HandleBatchGroupCount()
384 auto* dim = window.add_dimensions(); in HandleBatchGroupCount()
531 auto new_dim = new_window.add_dimensions(); in HandleConvolution()
646 auto window_dim = window.add_dimensions(); in HandleConvolution()
Dshape_inference_test.cc88 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
89 *window_.add_dimensions() = dim; in SelectAndScatterShapeInferenceTest()
324 *window.add_dimensions() = dim; in TEST_F()
325 *window.add_dimensions() = dim; in TEST_F()
424 auto dim0 = window.add_dimensions(); in TEST_F()
425 auto dim1 = window.add_dimensions(); in TEST_F()
469 auto dim0 = window.add_dimensions(); in TEST_F()
477 auto dim1 = window.add_dimensions(); in TEST_F()
515 auto dim0 = window.add_dimensions(); in TEST_F()
523 auto dim1 = window.add_dimensions(); in TEST_F()
[all …]
Dhlo_instructions.cc649 proto.add_dimensions(all_gather_dimension_); in ToProto()
743 proto.add_dimensions(*split_dimension_); in ToProto()
836 proto.add_dimensions(dimension); in ToProto()
874 proto.add_dimensions(dimension); in ToProto()
916 proto.add_dimensions(dimension); in ToProto()
961 proto.add_dimensions(dimension); in ToProto()
1016 proto.add_dimensions(dimension); in ToProto()
1054 proto.add_dimensions(dimension); in ToProto()
1111 proto.add_dimensions(inferred_dimension_); in ToProto()
1157 proto.add_dimensions(dimension); in ToProto()
[all …]
Dconvolution_4d_expander.cc85 *new_window.add_dimensions() = instruction->window().dimensions(i); in ExpandInstruction()
Dreshape_mover.cc220 shape->add_dimensions(i); in PerformSinkReshapeOrTranspose()
/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.cc74 auto* dim = padding_config.add_dimensions(); in Compile()
Dslice_op.cc155 auto* dims = padding_config.add_dimensions(); in Compile()
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dmlir_hlo_builder_test.cc162 auto* dims0 = padding_config.add_dimensions(); in TEST_F()
166 auto* dims1 = padding_config.add_dimensions(); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dpooling.cc56 auto dims = padding_config.add_dimensions(); in AvgPoolDivideByCountWithGeneralPadding()
96 padding_config.add_dimensions(); in MakeSpatialPaddingConfig()
/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()

123