/external/tensorflow/tensorflow/core/kernels/ |
D | concat_lib_gpu.cc | 40 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);
|
D | regex_full_match_op.cc | 50 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()
|
D | as_string_op.cc | 119 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()
|
D | stateful_random_ops.cc | 110 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()
|
D | unicode_script_op.cc | 35 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()
|
D | string_to_hash_bucket_op.h | 45 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()
|
D | regex_replace_op.cc | 49 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()
|
D | segment_reduction_ops.cc | 114 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 …]
|
D | pack_op.cc | 104 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()
|
D | decode_compressed_op.cc | 93 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()
|
D | string_to_number_op.cc | 49 auto output_flat = output_tensor->flat<OutputType>(); in Compute() local 55 &output_flat(i)), in Compute()
|
D | string_to_hash_bucket_op.cc | 42 auto output_flat = output_tensor->flat<int64>(); in Compute() local 51 output_flat(i) = static_cast<int64>(bucket_id); in Compute()
|
D | quantized_batch_norm_op.cc | 44 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()
|
D | concat_lib.h | 56 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat); 64 Tensor* output, typename TTypes<T, 2>::Tensor* output_flat);
|
D | random_op.cc | 203 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()
|
D | string_join_op.cc | 63 auto output_flat = output_tensor->flat<string>(); in Compute() local 70 output_flat(i) = str_util::Join(strings, separator_.c_str()); in Compute()
|
D | concat_op.cc | 156 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()
|
D | mfcc_op.cc | 77 float* output_flat = output_tensor->flat<float>().data(); in Compute() local 93 output_flat + in Compute()
|
D | spectrogram_op.cc | 71 auto output_flat = output_tensor->flat<float>().data(); in Compute() local 76 output_flat + (channel * output_height * output_width); in Compute()
|
D | sparse_softmax_op.cc | 81 typename TTypes<T>::Flat output_flat = output_values->flat<T>(); in Compute() local 117 output_flat.data() + count, group_size); in Compute()
|
D | list_kernels.h | 166 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/ |
D | zero_out_op_kernel_2.cc | 72 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/ |
D | scatter_nd_op_test.py | 58 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/ |
D | decode_video_op.cc | 90 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/ |
D | mfcc.cc | 116 float* output_flat = GetTensorData<float>(output); in Eval() local 131 float* output_data = output_flat + in Eval()
|