Home
last modified time | relevance | path

Searched refs:in_d (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h56 for (int in_d = 0; in_d < input_depth; ++in_d) { in SpaceToDepth() local
58 in_d + ((in_h % block_size) * block_size + in_w % block_size) * in SpaceToDepth()
64 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in SpaceToDepth()
Dconv3d.h65 const int in_d = in_d_origin + params.dilation_depth * filter_d; in Conv3D() local
76 (in_y < input_height) && (in_d >= 0) && in Conv3D()
77 (in_d < input_depth); in Conv3D()
86 input_shape, batch, in_d, in_y, in_x, in_channel)]; in Conv3D()
Ddepth_to_space.h57 const int in_d = in DepthToSpace() local
65 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in DepthToSpace()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_slice.cc76 for (int in_d = begin[3]; in_d <= end[3]; ++in_d) { in Slice() local
78 input_data[Offset(padded_shape, {in_b, in_h, in_w, in_d})]; in Slice()
/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op.cc167 const int in_d = d + offset_d; in operator ()() local
168 output(b, h, w, d) = input(b, in_h, in_w, in_d); in operator ()()
Ddepthwise_conv_grad_op.cc480 for (int in_d = 0; in_d < args.in_depth; ++in_d) { in DepthwiseConvBackpropInputReference() local
483 const int out_d_start = in_d * args.depth_multiplier; in DepthwiseConvBackpropInputReference()
510 (in_d + args.in_depth * (f_c + args.filter_cols * f_r)); in DepthwiseConvBackpropInputReference()
518 in_d + in DepthwiseConvBackpropInputReference()
995 const int in_d = out_d / args.depth_multiplier; in DepthwiseConvBackpropFilterReference() local
1012 in_d + in DepthwiseConvBackpropFilterReference()
1018 (in_d + args.in_depth * (f_c + args.filter_cols * f_r)); in DepthwiseConvBackpropFilterReference()
Ddepthtospace_op_gpu.cu.cc54 const int in_d = d + offset_d; in D2S_NHWC() local
56 in_d + input_depth * (in_w + input_width * (in_h + input_height * b)); in D2S_NHWC()
/external/webrtc/rtc_base/strings/
Djson_unittest.cc28 static Json::Value in_d(1.2); variable
139 EXPECT_TRUE(GetDoubleFromJson(in_d, &out)); in TEST()
/external/vixl/test/aarch64/
Dtest-assembler-sve-aarch64.cc3006 int in_d[] = {1, 1}; in TEST_SVE() local
3011 Initialise(&masm, p3.VnD(), in_d); in TEST_SVE()
3042 ASSERT_EQUAL_SVE(in_d, p3.VnD()); in TEST_SVE()
3347 int in_d[] = {1, 1}; in TEST_SVE() local
3352 Initialise(&masm, p3.VnD(), in_d); in TEST_SVE()
5131 uint64_t in_d[] = {0x8000000180018181, 0x7fffffff7fff7f7f, in TEST_SVE() local
5145 IntArithHelper(config, fn, kDRegSize, in_d, in_d, add_exp_d); in TEST_SVE()
5158 IntArithHelper(config, fn, kDRegSize, in_d, in_d, sqadd_exp_d); in TEST_SVE()
5171 IntArithHelper(config, fn, kDRegSize, in_d, in_d, uqadd_exp_d); in TEST_SVE()
10887 int64_t in_d[] = {1, 10, 10000, 1000000}; in TEST_SVE() local
[all …]