Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu.cc40 typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPUCall() argument
65 inputs_flat[0]->dimension(1), output_flat); in ConcatGPUCall()
75 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat) { in ConcatGPU() argument
78 ConcatGPUSlice<T, int32>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
80 ConcatGPUSlice<T, int64>(c->eigen_gpu_device(), inputs_flat, output_flat); in ConcatGPU()
87 ConcatGPUCall<T, int32>(c, inputs_flat, output_flat); in ConcatGPU()
89 ConcatGPUCall<T, int64>(c, inputs_flat, output_flat); in ConcatGPU()
99 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
Dregex_full_match_op.cc50 auto output_flat = output_tensor->flat<bool>(); in Compute() local
52 output_flat(i) = RE2::FullMatch(input_flat(i), match); in Compute()
79 auto output_flat = output_tensor->flat<bool>(); in Compute() local
81 output_flat(i) = RE2::FullMatch(input_flat(i), *re_); in Compute()
Das_string_op.cc119 auto output_flat = output_tensor->flat<string>(); in Compute() local
125 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \ in Compute()
139 output_flat(i) = (input_flat(i)) ? "true" : "false"; in Compute()
145 output_flat(i) = strings::Printf( in Compute()
152 output_flat(i) = strings::Printf( in Compute()
Dstateful_random_ops.cc110 auto output_flat = output->flat<T>(); in ComputeImpl() local
113 output_flat.size(), output_flat.data())); in ComputeImpl()
166 auto output_flat = output->flat<T>(); in Compute() local
167 auto data = output_flat.data(); in Compute()
168 for (int64 i = 0; i < output_flat.size(); ++i) { in Compute()
Dunicode_script_op.cc35 auto output_flat = output_tensor->flat<int32>(); in Compute() local
41 output_flat(i) = script_code; in Compute()
43 output_flat(i) = -1; in Compute()
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()
Dregex_replace_op.cc49 auto output_flat = output_tensor->flat<string>(); in InternalCompute() local
50 for (size_t i = 0; i < output_flat.size(); ++i) { in InternalCompute()
52 RE2::GlobalReplace(&output_flat(i), match, rewrite); in InternalCompute()
54 RE2::Replace(&output_flat(i), match, rewrite); in InternalCompute()
Dsegment_reduction_ops.cc114 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
164 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape); in Compute()
168 T* out_slice_ptr = &output_flat(out_index, 0); in Compute()
284 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local
289 output_flat); in ComputeAsync()
493 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
496 data.NumElements(), data_ptr, output_flat); in Compute()
694 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
729 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape); in Compute()
733 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()
Dconcat_lib.h56 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
64 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
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()
252 auto output_flat = output->flat<IntType>(); in Compute() local
257 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()
258 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.cc156 auto output_flat = output->shaped<T, 2>({inputs_flat_dim0, output_dim1}); in Compute() local
159 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
165 ConcatSYCL<T>(c->eigen_sycl_device(), inputs_flat, &output_flat); in Compute()
169 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.cc81 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local
117 output_flat.data() + count, group_size); in Compute()
Dlist_kernels.h166 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
170 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
174 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
479 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
483 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
487 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
655 auto output_flat = output->shaped<T, 2>({1, output->NumElements()}); in Compute() local
659 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
663 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/lite/kernels/
Dmfcc.cc116 float* output_flat = GetTensorData<float>(output); in Eval() local
131 float* output_data = output_flat + in Eval()

12