/external/ruy/ruy/ |
D | matrix.h | 39 void set_stride(int val) { stride_ = val; } in set_stride() function 173 layout->set_stride(order == Order::kColMajor ? rows : cols); in MakeSimpleLayout()
|
D | matrix_test.cc | 31 layout.set_stride(789); in TEST()
|
D | test.h | 493 layout->set_stride(min_stride); 495 layout->set_stride(min_stride + 1);
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | convolution_group_converter.cc | 296 window_dim->set_stride(batch_group_count - 1); in HandleBatchGroupCount() 390 dim->set_stride(batch_group_count); in HandleBatchGroupCount() 393 dim->set_stride(1); in HandleBatchGroupCount() 536 new_dim->set_stride(1); in HandleConvolution() 649 window_dim->set_stride(group_count - 1); in HandleConvolution()
|
D | transpose_folding_test.cc | 239 dim->set_stride(1); in TEST_F() 296 dim->set_stride(1); in TEST_F() 359 dim->set_stride(1); in TEST_F() 427 dim->set_stride(1); in TEST_F()
|
D | hlo_evaluator_test.cc | 170 dim.set_stride(stride); in ReduceWindowMaxIotaTest() 984 dim.set_stride(1); in TEST_P() 1047 dim.set_stride(1); in TEST_P() 1118 dim.set_stride(1); in TEST_P() 1197 dim.set_stride(1); in TEST_P() 1269 dim.set_stride(1); in TEST_P() 1333 dim.set_stride(1); in TEST_P() 1399 dim.set_stride(1); in TEST_P() 1406 dim.set_stride(3); in TEST_P() 1465 dim.set_stride(1); in TEST_P() [all …]
|
D | shape_inference_test.cc | 83 dim.set_stride(2); in SelectAndScatterShapeInferenceTest() 319 dim.set_stride(2); in TEST_F() 427 dim0->set_stride(2); in TEST_F() 433 dim1->set_stride(1); in TEST_F() 471 dim0->set_stride(3); in TEST_F() 479 dim1->set_stride(1); in TEST_F() 517 dim0->set_stride(3); in TEST_F() 525 dim1->set_stride(1); in TEST_F() 562 dim0->set_stride(1); in TEST_F() 566 dim1->set_stride(2); in TEST_F() [all …]
|
D | dot_as_convolution_util.cc | 119 wd->set_stride(std::max<int64>(1, wd->size() - 1)); in CreateShardedConvForDotGeneralConvolution()
|
D | dynamic_dimension_inference_test.cc | 946 batch_dim->set_stride(1); in TEST_F() 956 dim->set_stride(2); in TEST_F() 996 batch_dim->set_stride(1); in TEST_F() 1006 dim->set_stride(2); in TEST_F()
|
D | space_to_batch_converter.cc | 1444 new_win.mutable_dimensions(i)->set_stride( in Propagate() 2523 window_dim->set_stride(1); in PropagateOnBackpropFilterConv()
|
D | algebraic_simplifier_test.cc | 1730 dim->set_stride(1); in TEST_F() 1758 dim->set_stride(1); in TEST_F() 4073 f_dim_1->set_stride(options.row_stride); in TEST_F() 4080 f_dim_2->set_stride(options.col_stride); in TEST_F() 4381 dim->set_stride(1); in TEST_F() 4467 dim->set_stride(1); in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | window_util.cc | 34 dimension->set_stride(1); in MakeWindow() 48 dimension->set_stride(strides[nb]); in MakeWindow()
|
D | reference_util.cc | 448 dim.set_stride(kernel_stride.first); in ConvArray4DGeneralDimensionsDilated() 458 dim2.set_stride(kernel_stride.second); in ConvArray4DGeneralDimensionsDilated()
|
/external/llvm-project/polly/lib/External/isl/ |
D | isl_stride.c | 143 static isl_stat set_stride(struct isl_detect_stride_data *data, in set_stride() function 279 r = set_stride(data, stride, aff); in detect_stride()
|
D | isl_ast_build.c | 1301 static __isl_give isl_ast_build *set_stride(__isl_take isl_ast_build *build, in set_stride() function 1459 return set_stride(build, stride, offset); in isl_ast_build_detect_strides()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_pooling_gpu.cc | 84 pooling_desc.set_stride(dim_i, stride[i]); in Compute() 220 pooling_desc.set_stride(dim_i, stride[i]); in Compute()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | conv_canonicalization_test.cc | 41 dim->set_stride(1); in ConvCanonicalizationTest()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_rewriter.cc | 248 dim->set_stride(conv->window().dimensions(i).window_dilation()); in MatchBackwardFilter() 413 dim->set_stride(old_window.dimensions(i).base_dilation()); in MatchBackwardInput()
|
D | gpu_conv_rewriter_test.cc | 45 window_dim->set_stride(1); in GpuConvRewriterTest()
|
/external/ruy/example/ |
D | parametrized_example.cc | 90 layout->set_stride(stride ? stride : base_stride); in run()
|
/external/llvm-project/flang/include/flang/Evaluate/ |
D | variable.h | 167 Triplet &set_stride(Expr<SubscriptInteger> &&);
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | dot_handler.cc | 157 wd->set_stride(std::max<int64>(1, lhs_size - 1)); in GenNewWindow() 184 new_dim->set_stride(1); in GenNewWindow() 192 new_dim->set_stride(1); // std::max<int64>(1, 2 - 1) in GenNewWindow() 201 new_dim->set_stride(1); in GenNewWindow() 210 new_dim->set_stride(1); in GenNewWindow()
|
D | spmd_partitioner.cc | 1592 dim->set_stride(hlo->slice_strides(i)); in HandleSlice() 2057 dim->set_stride(1); in HandleReshape() 2112 dim->set_stride(1); in HandleReshape() 2633 dim->set_stride(1); in HandlePad()
|
D | convolution_handler.cc | 905 wd->set_stride(std::max<int64>(1, wd->size() - 1)); in CreateShardedConvConvolution()
|
/external/llvm-project/flang/lib/Evaluate/ |
D | variable.cpp | 66 Triplet &Triplet::set_stride(Expr<SubscriptInteger> &&expr) { in set_stride() function in Fortran::evaluate::Triplet
|