| /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
|