| /external/tensorflow/tensorflow/core/user_ops/ |
| fact.cc | 30 tensorflow::Tensor* output_tensor = nullptr; variable
32 0, tensorflow::TensorShape(), &output_tensor));
34 auto output = output_tensor->template scalar<string>();
|
| /external/tensorflow/tensorflow/user_ops/ |
| ackermann_op.cc | 35 Tensor* output_tensor = nullptr; variable
37 context->allocate_output(0, TensorShape(), &output_tensor));
38 auto output = output_tensor->scalar<string>();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| nth_element_op.h | 30 Tensor& output_tensor, int n);
|
| base64_ops.cc | 35 Tensor* output_tensor = nullptr; variable
37 &output_tensor));
40 auto output = output_tensor->flat<string>();
60 Tensor* output_tensor = nullptr; variable
62 &output_tensor));
65 auto output = output_tensor->flat<string>();
|
| string_to_number_op.cc | 45 Tensor* output_tensor = nullptr; variable
48 &output_tensor));
49 auto output_flat = output_tensor->flat<OutputType>();
|
| debug_ops.h | 165 Tensor* output_tensor; local
167 if (!context->allocate_output(0, shape, &output_tensor).ok()) {
233 Tensor* output_tensor; variable
253 OP_REQUIRES_OK(context, context->allocate_output(0, shape, &output_tensor));
254 output_tensor->vec<int64>()(0) = nan_count;
255 PublishTensor(*output_tensor); variable
276 Tensor* output_tensor; variable
356 OP_REQUIRES_OK(context, context->allocate_output(0, shape, &output_tensor));
357 output_tensor->vec<double>()(0) = static_cast<double>(is_initialized);
358 output_tensor->vec<double>()(1) = static_cast<double>(element_count)
380 PublishTensor(*output_tensor); variable
[all...] |
| nth_element_op.cc | 71 Tensor* output_tensor = nullptr; variable
73 context->allocate_output(0, out_shape, &output_tensor));
76 nthElementFunc(context, input_in, *output_tensor, n, reverse_);
88 Tensor& output_tensor, int n, bool reverse) {
90 T* output = output_tensor.flat<T>().data();
94 const int num_rows = output_tensor.NumElements();
|
| string_to_hash_bucket_op.h | 41 Tensor* output_tensor = nullptr; variable
44 &output_tensor));
45 auto output_flat = output_tensor->flat<int64>();
83 Tensor* output_tensor = nullptr; variable
86 &output_tensor));
87 auto output_flat = output_tensor->flat<int64>();
|
| mkl_tfconv_op.h | 103 Tensor* output_tensor = NULL; local
105 input_number, output_shape, &output_tensor));
106 CHECK_NOTNULL(output_tensor);
112 CHECK_EQ(input.CheckReorderToOpMem(output_tf_pd, output_tensor, &net),
117 CHECK(output_tensor->CopyFrom(input_tensor, output_shape));
162 Tensor* output_tensor = NULL; local
164 &output_tensor));
173 static_cast<void*>(const_cast<T*>(output_tensor->flat<T>().data()));
|
| bitcast_op.cc | 61 Tensor output_tensor; variable
63 output_tensor.UnsafeCopyFromInternal(input_tensor, output_data_type_,
65 context->set_output(0, output_tensor);
|
| string_join_op.cc | 60 Tensor* output_tensor = nullptr; variable
62 &output_tensor));
63 auto output_flat = output_tensor->flat<string>();
|
| string_to_hash_bucket_op.cc | 38 Tensor* output_tensor = nullptr; variable
41 &output_tensor));
42 auto output_flat = output_tensor->flat<int64>();
|
| decode_raw_op.cc | 55 Tensor* output_tensor = nullptr; variable
57 &output_tensor));
67 Tensor* output_tensor = nullptr; variable
69 context, context->allocate_output("output", out_shape, &output_tensor));
70 auto out = output_tensor->flat_inner_dims<T>();
|
| bucketize_op.cc | 66 Tensor* output_tensor = nullptr; variable
68 &output_tensor));
69 auto output = output_tensor->template flat<int32>();
|
| fact_op.cc | 85 Tensor* output_tensor = nullptr; local
87 context, context->allocate_output(0, TensorShape({}), &output_tensor));
88 auto output = output_tensor->template scalar<string>();
|
| /external/tensorflow/tensorflow/contrib/input_pipeline/kernels/ |
| input_pipeline_kernels.cc | 36 Tensor* output_tensor = nullptr; variable
38 &output_tensor));
51 output_tensor->scalar<string>()() = list_flat(*pos);
|
| /external/tensorflow/tensorflow/examples/adding_an_op/ |
| cuda_op_kernel.cc | 43 Tensor* output_tensor = nullptr; variable
45 &output_tensor));
46 auto output = output_tensor->template flat<int32>();
|
| zero_out_op_kernel_1.cc | 46 Tensor* output_tensor = nullptr; variable
48 &output_tensor));
49 auto output = output_tensor->template flat<int32>();
|
| zero_out_op_kernel_3.cc | 53 Tensor* output_tensor = nullptr; variable
55 &output_tensor));
56 auto output = output_tensor->template flat<int32>();
|
| /external/tensorflow/tensorflow/tools/ci_build/builds/user_ops/ |
| cuda_op_kernel.cc | 43 Tensor* output_tensor = NULL; variable
45 &output_tensor));
46 auto output = output_tensor->template flat<int32>();
|
| zero_out_op_kernel_1.cc | 46 Tensor* output_tensor = NULL; variable
48 &output_tensor));
49 auto output = output_tensor->template flat<int32>();
|
| /external/tensorflow/tensorflow/contrib/periodic_resample/python/kernel_tests/ |
| periodic_resample_op_test.py | 36 output_tensor = input_tensor.reshape((6, 2))
41 self.assertAllEqual(result, output_tensor)
47 output_tensor = input_tensor.reshape((6, 2))[:-1]
52 self.assertAllEqual(result, output_tensor)
58 output_tensor = numpy.array([[[0], [2], [4], [6]], [[1], [3], [5], [7]],
62 # NOTE: output_tensor != input_tensor.reshape((4, 4, -1))
70 self.assertAllEqual(result, output_tensor)
76 output_tensor = numpy.array(
87 # NOTE: output_tensor != input_tensor.reshape((4, 4, 4, -1))
91 self.assertAllEqual(result, output_tensor)
[all...] |
| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| memory_stats_ops.cc | 30 Tensor* output_tensor = nullptr; variable
32 context, context->allocate_output(0, TensorShape({}), &output_tensor));
33 output_tensor->scalar<int64>()() = ExtractAllocatorStats(allocator_stats);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| fractional_max_pool_op_test.py | 104 output_tensor = np.zeros(shape=output_shape, dtype=input_tensor.dtype)
109 output_tensor[batch, :, :, channel] = self._MaxPoolAlongCols(
112 return output_tensor
383 output_tensor = nn_ops.max_pool(input_tensor, window_size,
385 output_data = output_tensor.eval()
388 output_tensor,
398 output_tensor,
421 output_tensor = nn_ops.max_pool(input_tensor, window_size,
423 output_data = output_tensor.eval()
426 output_tensor,
[all...] |
| /external/tensorflow/tensorflow/contrib/boosted_trees/lib/models/ |
| multiple_additive_trees_test.cc | 58 auto output_tensor = AsTensor<float>({9.0f, 23.0f}, {2, 1}); local
59 auto output_matrix = output_tensor.matrix<float>();
93 auto output_tensor = AsTensor<float>({0.0f, 0.0f}, {2, 1}); local
94 auto output_matrix = output_tensor.matrix<float>();
169 auto output_tensor = AsTensor<float>({0.0f, 0.0f, 0.0f, 0.0f}, {2, 2}); local
170 auto output_matrix = output_tensor.matrix<float>();
254 auto output_tensor = local
256 auto output_matrix = output_tensor.matrix<float>();
|
