| /external/tensorflow/tensorflow/core/kernels/ |
| D | edit_distance_op.cc | 159 auto output_t = output->flat<float>(); in Compute() local
160 output_t.setZero(); in Compute()
188 output_t(loc) = in Compute()
190 if (normalize_) output_t(loc) /= truth_seq.size(); in Compute()
197 output_t(loc) = hypothesis_seq.size(); in Compute()
198 if (normalize_ && output_t(loc) != 0.0f) { in Compute()
199 output_t(loc) = std::numeric_limits<float>::infinity(); in Compute()
205 output_t(loc) = (normalize_) ? 1.0 : truth_seq.size(); in Compute()
215 output_t(loc) = hypothesis_seq.size(); in Compute()
216 if (normalize_ && output_t(loc) != 0.0f) { in Compute()
[all …]
|
| D | searchsorted_op.cc | 99 Tensor* output_t; in Compute() local
100 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
102 if (output_t->dtype() == DT_INT32) { in Compute()
111 auto output = output_t->template flat<OutType>(); in Compute()
140 Tensor* output_t; in Compute() local
141 OP_REQUIRES_OK(ctx, ctx->allocate_output(0, values_t.shape(), &output_t)); in Compute()
143 if (output_t->dtype() == DT_INT32) { in Compute()
152 auto output = output_t->template flat<OutType>(); in Compute()
|
| D | matching_files_op.cc | 52 Tensor* output_t = nullptr; in Compute() local
55 &output_t)); in Compute()
56 auto output = output_t->vec<tstring>(); in Compute()
|
| D | population_count_op.cc | 47 Tensor* output_t; in Compute() local
48 OP_REQUIRES_OK(c, c->allocate_output(0, input_t.shape(), &output_t)); in Compute()
51 auto output = output_t->flat<uint8>(); in Compute()
|
| D | compare_and_bitpack_op.cc | 63 Tensor* output_t; in Compute() local
64 OP_REQUIRES_OK(c, c->allocate_output(0, output_shape, &output_t)); in Compute()
68 auto output = output_t->flat_inner_dims<uint8>(); in Compute()
|
| D | shape_ops.h | 178 Tensor output_t; in Compute() local
179 if (!output_t.CopyFrom(input_t, output_shape)) { in Compute()
187 ctx->set_output(0, std::move(output_t)); in Compute()
|
| D | one_hot_op.cc | 114 auto output_t = in Compute() local
119 off_value_t, &output_t); in Compute()
|
| D | slice_op.cc | 174 auto output_t = result->tensor<T, 2>(); in Compute() local
186 port::prefetch<port::PREFETCH_HINT_T0>(&output_t(i + 1, 0)); in Compute()
190 memcpy(&output_t(i, 0), &input_t(row, col_begin), in Compute()
|
| D | bincount_op.cc | 189 Tensor* output_t; in Compute() local
191 ctx->allocate_output(0, TensorShape({size}), &output_t)); in Compute()
192 auto output = output_t->flat<T>(); in Compute()
|
| D | xsmm_conv2d_test.cc | 136 LIBXSMM_VLA_DECL(4, float, output_t, output + (pad_w_out * ofwp + pad_h_out), in naive_conv_fp()
153 LIBXSMM_VLA_ACCESS(4, output_t, img, ofm, oj, oi, nOfm, ofhp, in naive_conv_fp()
|
| D | list_kernels.cc | 496 auto output_t = output->flat<Variant>(); in Compute() local
528 TensorList* out = output_t(i).get<TensorList>(); in Compute()
535 output_t(i) = std::move(out); in Compute()
|
| D | cudnn_rnn_ops.cc | 1135 Tensor* output_t = nullptr; in Compute() local
1136 OP_REQUIRES_OK(context, context->allocate_output(0, {1}, &output_t)); in Compute()
1137 *output_t->template flat<Index>().data() = params_size; in Compute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | cond_test.py | 56 output_t = f()
57 self.assertAllEqual([5.], self.evaluate(output_t))
76 output_t = f()
77 self.assertAllEqual([10.], self.evaluate(output_t))
96 output_t, = xla.compile(f)
97 self.assertAllEqual([5.], self.evaluate(output_t))
239 output_t = f()
240 self.assertAllEqual([10.], self.evaluate(output_t))
261 output_t, = xla.compile(f)
262 self.assertAllEqual([10.], self.evaluate(output_t))
|
| /external/tensorflow/tensorflow/core/kernels/boosted_trees/ |
| D | quantile_ops.cc | 185 Tensor* output_t; in Compute() local
191 &output_t)); in Compute()
192 auto output = output_t->matrix<float>(); in Compute()
246 Tensor* output_t; in Compute() local
250 &output_t)); in Compute()
251 auto output = output_t->matrix<float>(); in Compute()
535 Tensor* output_t = nullptr; in Compute() local
538 feature_idx, TensorShape({num_values}), &output_t)); in Compute()
539 auto output = output_t->flat<int32>(); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/tensor_forest/ |
| D | resource_ops.cc | 107 Tensor* output_t = nullptr; in Compute() local
109 context->allocate_output(0, TensorShape(), &output_t)); in Compute()
110 output_t->scalar<int32>()() = decision_tree_resource->get_size(); in Compute()
|
| /external/harfbuzz_ng/util/ |
| D | shape-consumer.hh | 34 template <typename output_t>
94 output_t output;
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | softmax_op.cc | 81 Tensor output_t(cpu_allocator(), DT_VARIANT, TensorShape({})); in Compute() local
82 output_t.scalar<Variant>()() = std::move(output_matrix); in Compute()
83 ctx->set_output(0, output_t); in Compute()
|
| D | transpose_op.cc | 115 Tensor output_t(cpu_allocator(), DT_VARIANT, TensorShape({})); in Compute() local
116 output_t.scalar<Variant>()() = std::move(output_matrix); in Compute()
117 ctx->set_output(0, output_t); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | image_ops.cc | 98 Tensor* output_t; in DoImageProjectiveTransformOp() local
103 &output_t)); in DoImageProjectiveTransformOp()
104 auto output = output_t->tensor<T, 4>(); in DoImageProjectiveTransformOp()
|
| /external/deqp/external/vulkancts/modules/vulkan/pipeline/ |
| D | vktPipelineCreationCacheControlTests.cpp | 383 typename output_t = vector<move_t>>
384 output_t wrapHandles(const DeviceInterface& vk, in wrapHandles()
393 auto output = output_t{}; in wrapHandles()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | dataset_utils.cc | 695 auto output_t = output->flat_outer_dims<type>(); \ in CopyPartialBatch()
698 output_t.template chip<0>(i) = value_t.template chip<0>(i); \ in CopyPartialBatch()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 1205 output_t = output.stack()
1207 self.evaluate(output_t), [-5, -4, -3, -2, -1, 0, 1, 4, 9, 16])
1234 output_t = f()
1235 self.assertAllEqual(output_t, [-5, -4, -3, -2, -1, 0, 1, 4, 9, 16])
|