Home
last modified time | relevance | path

Searched refs:output_flat (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.cc54 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall() argument
79 inputs_flat[0]->dimension(1), output_flat); in ConcatGPUCall()
89 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU() argument
92 ConcatGPUSlice<T, int32>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
94 ConcatGPUSlice<T, int64>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
101 ConcatGPUCall<T, int32>(c, inputs_flat, output_flat); in ConcatGPU()
103 ConcatGPUCall<T, int64>(c, inputs_flat, output_flat); in ConcatGPU()
113 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
Dstring_to_hash_bucket_op.h45 auto output_flat = output_tensor->flat<int64>(); in Compute() local
54 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
87 auto output_flat = output_tensor->flat<int64>(); in Compute() local
96 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
Das_string_op.cc116 auto output_flat = output_tensor->flat<string>(); in Compute() local
122 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \ in Compute()
135 output_flat(i) = (input_flat(i)) ? "true" : "false"; in Compute()
141 output_flat(i) = strings::Printf( in Compute()
Dsegment_reduction_ops.cc112 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
162 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape); in Compute()
166 T* out_slice_ptr = &output_flat(out_index, 0); in Compute()
282 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local
287 output_flat); in ComputeAsync()
489 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
492 data.NumElements(), data_ptr, output_flat); in Compute()
685 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
720 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape); in Compute()
724 auto out = output_flat.template chip<0>(out_index); in Compute()
[all …]
Dpack_op.cc104 auto output_flat = in Compute() local
117 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
123 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
127 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
Ddecode_compressed_op.cc93 auto output_flat = output_tensor->flat<string>(); in Compute() local
96 output_flat(i) = bytes_flat(i); in Compute()
112 output_flat(i) = output_string; in Compute()
Dstring_to_number_op.cc49 auto output_flat = output_tensor->flat<OutputType>(); in Compute() local
55 &output_flat(i)), in Compute()
Dstring_to_hash_bucket_op.cc42 auto output_flat = output_tensor->flat<int64>(); in Compute() local
51 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
Dquantized_batch_norm_op.cc44 auto output_flat = output->flat<T2>(); in ReferenceBatchNorm() local
81 output_flat(input_index) = in ReferenceBatchNorm()
107 auto output_flat = output->flat<T2>(); in FixedPointBatchNorm() local
156 output_flat(input_index) = output_value; in FixedPointBatchNorm()
Drandom_op.cc203 auto output_flat = output->flat<T>(); in Compute() local
208 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()
209 output_flat.data(), output_flat.size(), Distribution()); in Compute()
248 auto output_flat = output->flat<IntType>(); in Compute() local
253 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()
254 output_flat.data(), output_flat.size(), dist); in Compute()
Dstring_join_op.cc63 auto output_flat = output_tensor->flat<string>(); in Compute() local
70 output_flat(i) = str_util::Join(strings, separator_.c_str()); in Compute()
Dconcat_op.cc134 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
137 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
143 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
147 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
Dmfcc_op.cc77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
93 output_flat + in Compute()
Dspectrogram_op.cc71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local
76 output_flat + (channel * output_height * output_width); in Compute()
Dsparse_softmax_op.cc78 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
114 output_flat.data() + count, group_size); in Compute()
Dlist_kernels.h117 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
121 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
125 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
Droll_op.cc289 auto output_flat = output->flat<T>().data(); in Compute() local
295 input_flat, output_flat, threshold, dim_range, isd); in Compute()
299 output_flat, threshold, dim_range); in Compute()
Dreduce_join_op.cc159 auto output_flat = output_tensor->flat<string>(); in Compute() local
175 output_flat(output_index) = in Compute()
Dconcat_lib.h55 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
Dquantized_concat_op.cc218 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
222 &output_flat); in Compute()
Dmkl_concat_op.cc147 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
148 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
225 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
226 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
/external/tensorflow/tensorflow/examples/adding_an_op/
Dzero_out_op_kernel_2.cc72 auto output_flat = output->template flat<T>(); in Compute() local
77 output_flat(i) = T(0); in Compute()
81 if (N > 0) output_flat(0) = input(0); in Compute()
/external/tensorflow/tensorflow/compiler/tests/
Dscatter_nd_op_test.py58 output_flat = _FlatOuterDims(ref, ixdim + 1)
61 output_flat[ix_output] = op(output_flat[ix_output],
63 return output_flat.reshape(ref.shape)
/external/tensorflow/tensorflow/contrib/ffmpeg/
Ddecode_video_op.cc90 auto output_flat = output->flat<uint8>(); in Compute() local
91 std::copy_n(output_data.begin(), output_data.size(), &output_flat(0)); in Compute()
/external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
Dfisher_blocks_test.py142 output_flat = sess.run(utils.tensors_to_column(output)).ravel()
147 self.assertAllClose(output_flat, explicit)
232 output_flat = sess.run(utils.tensors_to_column(output)).ravel()
237 self.assertAllClose(output_flat, explicit)
540 output_flat = sess.run(utils.tensors_to_column(output)).ravel()
545 self.assertAllClose(output_flat, explicit)
845 output_flat = sess.run(utils.tensors_to_column(output)).ravel()
850 self.assertAllClose(output_flat, explicit)

12