| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops.cc | 182 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()() local
183 int h_beg = h_out * stride_rows - pad_top; in operator ()()
202 output(b, h_out, w_out, d) = cur_val; in operator ()()
297 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()() local
298 int h_beg = h_out * stride_rows - pad_top; in operator ()()
322 out_backprop(b, h_out, w_out, d); in operator ()()
417 for (int h_out = 0; h_out < output_rows; ++h_out) { in operator ()() local
418 int h_beg = h_out * stride_rows - pad_top; in operator ()()
442 out_backprop(b, h_out, w_out, d); in operator ()()
|
| D | dilation_ops_gpu.cu.cc | 50 const int h_out = out_idx3 % output_rows; in DilationKernel() local
52 int h_beg = h_out * stride_rows - pad_top; in DilationKernel()
90 const int h_out = out_idx3 % output_rows; in DilationBackpropInputKernel() local
92 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropInputKernel()
140 const int h_out = out_idx3 % output_rows; in DilationBackpropFilterKernel() local
142 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropFilterKernel()
|
| /external/OpenCL-CTS/test_conformance/spirv_new/ |
| D | test_op_function.cpp | 54 std::vector<cl_float> h_out(num); in test_function() local
55 err = clEnqueueReadBuffer(queue, in, CL_TRUE, 0, bytes, &h_out[0], 0, NULL, NULL); in test_function()
59 if (h_out[i] != -h_in[i]) { in test_function()
|
| D | test_op_loop_merge.cpp | 68 std::vector<T> h_out(num); in test_selection_merge() local
69 err = clEnqueueReadBuffer(queue, out, CL_TRUE, 0, bytes, &h_out[0], 0, NULL, NULL); in test_selection_merge()
73 if (h_out[i] != h_ref[i]) { in test_selection_merge()
|
| D | test_op_vector_extract.cpp | 38 std::vector<Ts> h_out(num); in test_extract() local
66 err = clEnqueueReadBuffer(queue, out, CL_TRUE, 0, out_bytes, &h_out[0], 0, NULL, NULL); in test_extract()
70 if (h_out[i] != h_in[i].s[k]) { in test_extract()
|
| D | test_linkage.cpp | 131 std::vector<cl_float> h_out(num); in TEST_SPIRV_FUNC() local
132 err = clEnqueueReadBuffer(queue, in, CL_TRUE, 0, bytes, &h_out[0], 0, NULL, NULL); in TEST_SPIRV_FUNC()
136 if (h_out[i] != -h_in[i]) { in TEST_SPIRV_FUNC()
|
| D | test_op_negate.cpp | 64 std::vector<Tv> h_out(num); in test_negation() local
65 err = clEnqueueReadBuffer(queue, in, CL_TRUE, 0, bytes, &h_out[0], 0, NULL, NULL); in test_negation()
69 if (h_out[i] != negate(h_in[i])) { in test_negation()
|
| D | test_op_vector_insert.cpp | 38 std::vector<Tv> h_out(num); in test_insert() local
78 err = clEnqueueReadBuffer(queue, out, CL_TRUE, 0, out_bytes, &h_out[0], 0, NULL, NULL); in test_insert()
87 if (h_out[i].s[j] != refVec.s[j]) { in test_insert()
|
| D | test_decorate.cpp | 319 std::vector<To> &h_out) in TEST_SATURATED_CONVERSION()
363 if (h_res[i] != h_out[i]) { in TEST_SATURATED_CONVERSION()
365 i, h_in[i], h_out[i], h_res[i]); in TEST_SATURATED_CONVERSION()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops.cc | 955 Tensor* h_out; in Compute() local
956 OP_REQUIRES_OK(ctx, ctx->allocate_output("h", batch_cell_shape, &h_out)); in Compute()
980 t == 0 ? *h_prev_tensor : slicer.OutputSlice(h_out, t - 1, "h_prev"); in Compute()
988 Tensor h_tensor = slicer.OutputSlice(h_out, t, "h_out"); in Compute()
1005 Tensor h_tensor = h_out->Slice(seq_len_max, timelen); in Compute()
1135 const Tensor* h_out = nullptr; in Compute() local
1136 OP_REQUIRES_OK(ctx, ctx->input("h", &h_out)); in Compute()
1245 t == 0 ? *h_prev_tensor : slicer.InputSlice(*h_out, t - 1, "h_prev"); in Compute()
|