Home
last modified time | relevance | path

Searched refs:OutputT (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dredux_functor.h38 template <typename InputT, typename AccumT, typename OutputT,
49 auto output_dims = output->template flat<OutputT>().dimensions(); in operator()
59 output->template flat<OutputT>() = in operator()
60 input.template flat<InputT>().template cast<OutputT>().reshape( in operator()
124 output->template flat<OutputT>() = in operator()
125 buffer.template cast<OutputT>().reshape(output_dims); in operator()
195 output->template flat<OutputT>() = in operator()
196 buf0.template cast<OutputT>().reshape(output_dims); in operator()
206 template <typename InputT, typename AccumT, typename OutputT,
218 auto output_dims = output->template flat<OutputT>().dimensions(); in operator()
[all …]
Dspacetobatch_functor.h88 using OutputT = typename std::conditional<B2S, const T, T>::type; member
108 typename TTypes<OutputT, NUM_BLOCK_DIMS + 2>::Tensor batch_tensor);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dquantize.h30 template <typename InputT, typename OutputT>
35 OutputT* output_data) { in AffineQuantize()
39 static constexpr int32_t min_val = std::numeric_limits<OutputT>::min(); in AffineQuantize()
40 static constexpr int32_t max_val = std::numeric_limits<OutputT>::max(); in AffineQuantize()
Ddequantize.h30 template <typename InputT, typename OutputT>
34 const RuntimeShape& output_shape, OutputT* output_data) { in Dequantize()
41 const OutputT result = static_cast<OutputT>(scale * (val - zero_point)); in Dequantize()
Dsoftmax.h66 template <typename InputT, typename OutputT>
69 const RuntimeShape& output_shape, OutputT* output_data) { in Softmax()
130 num_bits_over_unit + 31 - (sizeof(OutputT) * 8)); in Softmax()
134 static_cast<int32_t>(std::numeric_limits<OutputT>::min()); in Softmax()
136 output_data[i * depth + c] = static_cast<OutputT>(std::max( in Softmax()
138 static_cast<int32_t>(std::numeric_limits<OutputT>::max())), in Softmax()
139 static_cast<int32_t>(std::numeric_limits<OutputT>::min()))); in Softmax()
141 output_data[i * depth + c] = std::numeric_limits<OutputT>::min(); in Softmax()
/external/llvm-project/clang/test/SemaTemplate/
Ddeduction-crash.cpp150 template <typename InputT, typename OutputT>
153 OutputT evaluateSelectionRequirement(InputT &&Value) {
157 template <typename InputT, typename OutputT>
158 OutputT SourceSelectionRequirement<InputT, OutputT>::
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.h118 template <typename FuncT, typename InputT, typename OutputT>
122 DeviceMemory<OutputT> *output);
126 template <typename FuncT, typename InputT, typename OutputT>
129 DeviceMemory<OutputT> *output);
Drocm_fft.cc508 template <typename FuncT, typename InputT, typename OutputT>
511 DeviceMemory<OutputT> *output) { in DoFftInternal()
534 template <typename FuncT, typename InputT, typename OutputT>
538 DeviceMemory<OutputT> *output) { in DoFftWithDirectionInternal()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.h122 template <typename FuncT, typename InputT, typename OutputT>
126 DeviceMemory<OutputT> *output);
130 template <typename FuncT, typename InputT, typename OutputT>
133 DeviceMemory<OutputT> *output);
Dcuda_fft.cc457 template <typename FuncT, typename InputT, typename OutputT>
460 DeviceMemory<OutputT> *output) { in DoFftInternal()
480 std::is_same<OutputT, float>::value && input.size() > 0) { in DoFftInternal()
510 template <typename FuncT, typename InputT, typename OutputT>
514 DeviceMemory<OutputT> *output) { in DoFftWithDirectionInternal()
/external/guava/guava-gwt/src-super/com/google/common/util/concurrent/super/com/google/common/util/concurrent/
DAggregateFutureState.java24 abstract class AggregateFutureState<OutputT> extends AbstractFuture.TrustedFuture<OutputT> {
/external/guava/guava/src/com/google/common/util/concurrent/
DAggregateFutureState.java40 abstract class AggregateFutureState<OutputT> extends AbstractFuture.TrustedFuture<OutputT> {
DAggregateFuture.java42 abstract class AggregateFuture<InputT, OutputT> extends AggregateFutureState<OutputT> {
/external/guava/android/guava/src/com/google/common/util/concurrent/
DAggregateFutureState.java40 abstract class AggregateFutureState<OutputT> extends AbstractFuture.TrustedFuture<OutputT> {
DAggregateFuture.java42 abstract class AggregateFuture<InputT, OutputT> extends AggregateFutureState<OutputT> {