Home
last modified time | relevance | path

Searched refs:output_flat (Results 1 – 25 of 44) 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);
Das_string_op.cc141 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
147 output_flat(i) = strings::Printf((enc_str.c_str()), input_flat(i)); \ in Compute()
161 output_flat(i) = (input_flat(i)) ? "true" : "false"; in Compute()
167 output_flat(i) = input_flat(i).DebugString(); in Compute()
173 output_flat(i) = strings::Printf( in Compute()
180 output_flat(i) = strings::Printf( 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()
Dregex_full_match_op.cc54 auto output_flat = output_tensor->flat<bool>(); in Compute() local
56 output_flat(i) = RE2::FullMatch(input_flat(i), *regex); in Compute()
107 auto output_flat = output_tensor->flat<bool>(); in Compute() local
109 output_flat(i) = RE2::FullMatch(input_flat(i), *re_); in Compute()
Drandom_op.cc75 auto output_flat = output->flat<T>(); in Compute() local
80 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()
81 output_flat.data(), output_flat.size(), Distribution()); in Compute()
124 auto output_flat = output->flat<IntType>(); in Compute() local
129 generator_.ReserveRandomOutputs(output_flat.size(), 256), in Compute()
130 output_flat.data(), output_flat.size(), dist); in Compute()
Dregex_replace_op.cc51 auto output_flat = output_tensor->flat<tstring>(); in InternalCompute() local
52 for (size_t i = 0; i < output_flat.size(); ++i) { in InternalCompute()
55 string buf = output_flat(i); in InternalCompute()
61 output_flat(i) = std::move(buf); in InternalCompute()
Dunsorted_segment_join_op.cc118 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
141 if (output_flat(output_index).length() != 0) in Compute()
142 output_flat(output_index).append(separator_.c_str()); in Compute()
143 output_flat(output_index).append(flat_input(offset)); in Compute()
Dstateful_random_ops.cc137 auto output_flat = output->flat<T>(); in StatefulRandomCompute() local
140 output_flat.size(), output_flat.data())); in StatefulRandomCompute()
290 auto output_flat = output->flat<T>(); in Compute() local
293 output_flat); in Compute()
296 ctx->eigen_device<Device>(), output_flat, in Compute()
333 auto output_flat = output->flat<T>(); in Compute() local
334 auto data = output_flat.data(); in Compute()
335 for (int64 i = 0; i < output_flat.size(); ++i) { in Compute()
Dsegment_reduction_ops_impl.h123 auto output_flat = output->flat_outer_dims<T>(); in Compute() local
173 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape); in Compute()
177 T* out_slice_ptr = &output_flat(out_index, 0); in Compute()
295 auto output_flat = output->flat_outer_dims<T>(); in ComputeAsync() local
300 output_flat); in ComputeAsync()
414 auto output_flat = output->flat_outer_dims<T>();
417 output_flat);
511 auto output_flat = output->flat_outer_dims<T>();
552 gap_slice(&output_flat(uninitialized_index, 0), gap_slice_shape);
556 auto out = output_flat.template chip<0>(out_index);
[all …]
Ddecode_compressed_op.cc93 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
96 output_flat(i) = bytes_flat(i); in Compute()
112 output_flat(i) = std::move(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()
Dconcat_lib.h57 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
65 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
Dstring_to_hash_bucket_op.cc41 auto output_flat = output_tensor->flat<int64>(); in Compute() local
50 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
Dstring_to_hash_bucket_fast_op.h45 auto output_flat = output_tensor->flat<int64>(); in Compute() local
54 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
Dpack_op.cc91 auto output_flat = in Compute() local
111 ConcatGPU<T>(c, inputs_flat, output, &output_flat); in Compute()
115 ConcatCPU<T>(c->device(), inputs_flat, &output_flat); in Compute()
Dstring_to_hash_bucket_op.h52 auto output_flat = output_tensor->flat<int64>(); in Compute() local
61 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
Dsobol_op.cc159 auto output_flat = output->flat<T>(); in Compute() local
168 [&dim, &skip, &output_flat](const int start, const int end) { in Compute()
170 start, output_flat); in Compute()
Dstring_join_op.cc63 auto output_flat = output_tensor->flat<tstring>(); in Compute() local
70 output_flat(i) = absl::StrJoin(strings, separator_); 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()
Dmfcc_op.cc77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local
93 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/lite/python/testdata/
Ddouble_op.cc42 auto output_flat = output_tensor->flat<int32>(); in Compute() local
47 output_flat(i) = 2 * input_flat(i); in Compute()
/external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/
Dscatter_nd_ops_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/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
Dsentencepiece_detokenizer_op.cc59 auto output_flat = output_tensor->flat<tensorflow::tstring>(); in Compute() local
78 output_flat(i) = res.decoded; in Compute()
/external/tensorflow/tensorflow/compiler/tests/
Dscatter_nd_op_test.py59 output_flat = _FlatOuterDims(ref, ixdim + 1)
62 output_flat[ix_output] = op(output_flat[ix_output],
64 return output_flat.reshape(ref.shape)

12