Searched refs:input_data (Results 76 - 100 of 180) sorted by relevance

12345678

/external/tensorflow/tensorflow/core/kernels/
H A Dadjust_hue_op.cc211 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize}); variable
219 [channel_count, &input_data, &output_data, delta_h](
221 const float* p = input_data.data() + start_channel * kChannelSize;
268 const float* input_data = input->flat<float>().data(); variable
271 functor::AdjustHueGPU()(&device, number_of_elements, input_data, delta_h,
H A Dadjust_saturation_op.cc188 auto input_data = input->shaped<float, 2>({channel_count, kChannelSize}); variable
196 [channel_count, &input_data, &output_data, scale_h](
198 const float* p = input_data.data() + start_channel * kChannelSize;
234 const float* input_data = input->flat<float>().data(); variable
237 functor::AdjustSaturationGPU()(&device, number_of_elements, input_data,
H A Drandom_crop_op.cc95 typename TTypes<T, 3>::ConstTensor input_data(input.tensor<T, 3>());
102 input_data(y + offset_height, x + offset_width, c);
H A Dresize_area_op.cc152 typename TTypes<T, 4>::ConstTensor input_data(input.tensor<T, 4>());
183 ComputeLoop<3>(st, x_interps, input_data);
185 ComputeLoop<-1>(st, x_interps, input_data);
192 typename TTypes<T, 4>::ConstTensor input_data) {
214 const T* const input_ptr = input_data.data();
190 ComputeLoop(const ImageResizerState& st, const std::vector<CachedInterpolation>& x_interps, typename TTypes<T, 4>::ConstTensor input_data) argument
/external/tensorflow/tensorflow/examples/speech_commands/
H A Dgenerate_streaming_test_wav.py49 import input_data namespace
87 words_list = input_data.prepare_words_list(FLAGS.wanted_words.split(','))
91 audio_processor = input_data.AudioProcessor(
142 wanted_label = input_data.UNKNOWN_WORD_LABEL
160 input_data.save_wav_file(FLAGS.output_audio_file, output_audio,
/external/libjpeg-turbo/simd/
H A Djdsample-mmx.asm50 ; JSAMPARRAY input_data,
56 %define input_data(b) (b)+16 ; JSAMPARRAY input_data
81 mov esi, JSAMPARRAY [input_data(ebp)] ; input_data
179 add esi, byte SIZEOF_JSAMPROW ; input_data
203 ; JSAMPARRAY input_data,
209 %define input_data(b) (b)+16 ; JSAMPARRAY input_data
247 mov esi, JSAMPARRAY [input_data(ed
[all...]
H A Djdsample-sse2.asm50 ; JSAMPARRAY input_data,
56 %define input_data(b) (b)+16 ; JSAMPARRAY input_data
81 mov esi, JSAMPARRAY [input_data(ebp)] ; input_data
179 add esi, byte SIZEOF_JSAMPROW ; input_data
201 ; JSAMPARRAY input_data,
207 %define input_data(b) (b)+16 ; JSAMPARRAY input_data
245 mov esi, JSAMPARRAY [input_data(ed
[all...]
H A Djsimd_arm64.c378 JSAMPARRAY input_data, JSAMPARRAY output_data)
382 input_data, output_data);
387 JSAMPARRAY input_data, JSAMPARRAY output_data)
391 input_data, output_data);
413 JSAMPARRAY input_data,
421 JSAMPARRAY input_data,
445 JSAMPARRAY input_data,
453 JSAMPARRAY input_data,
377 jsimd_h2v2_downsample(j_compress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY output_data) argument
386 jsimd_h2v1_downsample(j_compress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY output_data) argument
411 jsimd_h2v2_upsample(j_decompress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) argument
419 jsimd_h2v1_upsample(j_decompress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) argument
443 jsimd_h2v2_fancy_upsample(j_decompress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) argument
451 jsimd_h2v1_fancy_upsample(j_decompress_ptr cinfo, jpeg_component_info *compptr, JSAMPARRAY input_data, JSAMPARRAY *output_data_ptr) argument
/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dflatten_atrous_test.cc43 Tensor input_data(DT_FLOAT, TensorShape({1, 3, 3, 2}));
45 &input_data, {.1f, .4f, .2f, .5f, .3f, .6f, -1.0f, -.4f, -.2f, -.5f,
48 Const(root.WithOpName("input_op"), Input::Initializer(input_data));
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbroadcast_test.cc175 std::vector<float> input_data(1025);
176 int64 r1_size = input_data.size();
177 std::iota(input_data.begin(), input_data.end(), 0.0f);
179 HloInstruction::CreateConstant(Literal::CreateR1<float>(input_data)));
194 yx(y, x) = input_data[x];
H A Dexhaustive_f32_elementwise_op_test.cc53 TF_ASSERT_OK_AND_ASSIGN(std::unique_ptr<GlobalData> input_data,
65 ComputeAndCompareR1<float>(&builder, expected_result, {input_data.get()},
H A Dconvolution_test.cc131 Array4D<T> input_data(1, 1, 1, 2);
132 input_data.FillWithYX(Array2D<T>({
141 {std::move(*Literal::CreateFromArray(input_data)),
162 Array4D<T> input_data(1, 1, 4, 4);
163 input_data.FillWithYX(Array2D<T>({
175 {std::move(*Literal::CreateFromArray(input_data)),
196 Array4D<T> input_data(1, 1, 4, 4);
197 input_data.FillWithYX(Array2D<T>({
210 {std::move(*Literal::CreateFromArray(input_data)),
232 Array4D<T> input_data(
[all...]
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
H A Ddecision-tree-resource.cc30 const std::unique_ptr<TensorDataSet>& input_data, int example,
48 node_evaluators_[current_id]->Decide(input_data, example);
29 TraverseTree( const std::unique_ptr<TensorDataSet>& input_data, int example, int32* leaf_depth, TreePath* path) const argument
H A Dsplit_collection_operators.h21 #include "tensorflow/contrib/tensor_forest/kernels/v4/input_data.h"
52 virtual void AddExample(const std::unique_ptr<TensorDataSet>& input_data,
59 const std::unique_ptr<TensorDataSet>& input_data,
H A Ddecision-tree-resource.h20 #include "tensorflow/contrib/tensor_forest/kernels/v4/input_data.h"
64 int32 TraverseTree(const std::unique_ptr<TensorDataSet>& input_data,
H A Dfertile-stats-resource.h21 #include "tensorflow/contrib/tensor_forest/kernels/v4/input_data.h"
68 const std::unique_ptr<TensorDataSet>& input_data,
/external/tensorflow/tensorflow/examples/learn/
H A Drandom_forest_mnist.py30 from tensorflow.examples.tutorials.mnist import input_data namespace
58 mnist = input_data.read_data_sets(FLAGS.data_dir, one_hot=False)
/external/tensorflow/tensorflow/examples/tutorials/mnist/
H A Dmnist_softmax_xla.py27 from tensorflow.examples.tutorials.mnist import input_data namespace
35 mnist = input_data.read_data_sets(FLAGS.data_dir)
101 default='/tmp/tensorflow/mnist/input_data',
/external/tensorflow/tensorflow/python/lib/core/
H A Dpy_func.cc313 PyObject** input_data = reinterpret_cast<PyObject**>(PyArray_DATA(input)); local
317 if (PyBytes_AsStringAndSize(input_data[i], &el, &el_size) == -1) {
319 el = PyUnicode_AsUTF8AndSize(input_data[i], &el_size);
322 if (PyUnicode_Check(input_data[i])) {
323 PyObject* unicode = PyUnicode_AsUTF8String(input_data[i]);
334 input_data[i]->ob_type->tp_name);
346 char* input_data = PyArray_BYTES(input); local
349 tflat(i) = string(input_data + i * el_size, el_size);
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dconcat_op.cc75 std::vector<xla::ComputationDataHandle> input_data;
91 input_data.push_back(ctx->builder()->Reshape(handle, {1}));
93 input_data.push_back(handle);
99 ctx->SetOutput(0, ctx->builder()->ConcatInDim(input_data, axis));
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
H A Dcudnn_rnn_ops_benchmark.py99 input_data = variables.Variable(
109 input_data=input_data,
115 [params, input_data, input_h, input_c])
H A Dcudnn_rnn_ops_test.py317 input_data = constant_op.constant(
327 input_data=input_data,
334 input_data=input_data,
434 input_data = array_ops.ones([seq_length, batch_size, input_size])
441 input_data=input_data,
448 input_data=input_data,
[all...]
/external/tensorflow/tensorflow/contrib/learn/python/learn/learn_io/
H A Ddata_feeder_test.py41 def _assert_raises(self, input_data):
43 data_feeder.DataFeeder(input_data, None, n_classes=0, batch_size=1)
45 def _assert_dtype(self, expected_np_dtype, expected_tf_dtype, input_data):
46 feeder = data_feeder.DataFeeder(input_data, None, n_classes=0, batch_size=1)
47 if isinstance(input_data, dict):
121 input_data = np.array([['str%d' % i for i in xrange(2)] for _ in xrange(2)])
122 self._assert_dtype(input_data.dtype, dtypes.string, input_data)
123 self._assert_dtype(input_data.dtype, dtypes.string,
124 self._wrap_dict(input_data))
[all...]
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dfractional_avg_pool_op_test.py466 input_data = self._GenerateRandomInputTensor(input_shape)
472 input_tensor = constant_op.constant(input_data, shape=input_shape)
490 x_init_value=input_data.reshape(input_shape),
503 input_data = self._GenerateRandomInputTensor(input_shape)
505 input_tensor = constant_op.constant(input_data, shape=input_shape)
523 x_init_value=input_data.reshape(input_shape),
529 input_data = self._GenerateRandomInputTensor(input_shape)
536 input_tensor = constant_op.constant(input_data, shape=input_shape)
552 x_init_value=input_data.reshape(input_shape),
/external/tensorflow/tensorflow/stream_executor/cuda/
H A Dcuda_dnn.cc1426 const DeviceMemory<T>& input_data,
1536 const DeviceMemory<T>& input_data,
1552 rnn_desc, input_desc, input_data, input_h_desc, input_h_data,
1610 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1628 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1653 const DeviceMemory<T>& input_data,
1676 rnn_desc, input_desc, input_data, input_h_desc, input_h_data,
1733 input_desc.handles() /*xDesc*/, input_data.opaque() /*x*/,
1831 const DeviceMemory<Eigen::half>& input_data,
1862 stream, cudnn_rnn_desc, cudnn_input_desc, input_data, cudnn_input_h_des
1423 ExtractAndCheckRnnForward( const CudnnRnnDescriptor& rnn_desc, const CudnnRnnSequenceTensorDescriptor& input_desc, const DeviceMemory<T>& input_data, const CudnnRnnStateTensorDescriptor& input_h_desc, const DeviceMemory<T>& input_h_data, const CudnnRnnStateTensorDescriptor& input_c_desc, const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, const CudnnRnnSequenceTensorDescriptor& output_desc, const DeviceMemory<T>& output_data, const CudnnRnnStateTensorDescriptor& output_h_desc, const DeviceMemory<T>& output_h_data, const CudnnRnnStateTensorDescriptor& output_c_desc, const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) argument
1533 DoRnnForwardImpl( Stream* stream, const CudnnRnnDescriptor& rnn_desc, const CudnnRnnSequenceTensorDescriptor& input_desc, const DeviceMemory<T>& input_data, const CudnnRnnStateTensorDescriptor& input_h_desc, const DeviceMemory<T>& input_h_data, const CudnnRnnStateTensorDescriptor& input_c_desc, const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, const CudnnRnnSequenceTensorDescriptor& output_desc, DeviceMemory<T>* output_data, const CudnnRnnStateTensorDescriptor& output_h_desc, DeviceMemory<T>* output_h_data, const CudnnRnnStateTensorDescriptor& output_c_desc, DeviceMemory<T>* output_c_data, bool is_training, ScratchAllocator* reserve_space_allocator, ScratchAllocator* workspace_allocator) argument
1650 DoRnnBackwardImpl( Stream* stream, const CudnnRnnDescriptor& rnn_desc, const CudnnRnnSequenceTensorDescriptor& input_desc, const DeviceMemory<T>& input_data, const CudnnRnnStateTensorDescriptor& input_h_desc, const DeviceMemory<T>& input_h_data, const CudnnRnnStateTensorDescriptor& input_c_desc, const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, const CudnnRnnSequenceTensorDescriptor& output_desc, const DeviceMemory<T>& output_data, const CudnnRnnStateTensorDescriptor& output_h_desc, const DeviceMemory<T>& output_h_data, const CudnnRnnStateTensorDescriptor& output_c_desc, const DeviceMemory<T>& output_c_data, const DeviceMemory<T>& output_backprop_data, const DeviceMemory<T>& output_h_backprop_data, const DeviceMemory<T>& output_c_backprop_data, DeviceMemory<T>* input_backprop_data, DeviceMemory<T>* input_h_backprop_data, DeviceMemory<T>* input_c_backprop_data, DeviceMemory<T>* params_backprop_data, DeviceMemory<uint8>* reserve_space_data, ScratchAllocator* workspace_allocator) argument
1828 DoRnnForward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<Eigen::half>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<Eigen::half>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<Eigen::half>& input_c_data, const DeviceMemory<Eigen::half>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, DeviceMemory<Eigen::half>* output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, DeviceMemory<Eigen::half>* output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, DeviceMemory<Eigen::half>* output_c_data, bool is_training, ScratchAllocator* reserve_space_allocator, ScratchAllocator* workspace_allocator) argument
1871 DoRnnForward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<float>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<float>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, DeviceMemory<float>* output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, DeviceMemory<float>* output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, DeviceMemory<float>* output_c_data, bool is_training, ScratchAllocator* reserve_space_allocator, ScratchAllocator* workspace_allocator) argument
1913 DoRnnForward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<double>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<double>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<double>& input_c_data, const DeviceMemory<double>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, DeviceMemory<double>* output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, DeviceMemory<double>* output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, DeviceMemory<double>* output_c_data, bool is_training, ScratchAllocator* reserve_space_allocator, ScratchAllocator* workspace_allocator) argument
1956 DoRnnBackward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<Eigen::half>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<Eigen::half>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<Eigen::half>& input_c_data, const DeviceMemory<Eigen::half>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, const DeviceMemory<Eigen::half>& output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, const DeviceMemory<Eigen::half>& output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, const DeviceMemory<Eigen::half>& output_c_data, const DeviceMemory<Eigen::half>& output_backprop_data, const DeviceMemory<Eigen::half>& output_h_backprop_data, const DeviceMemory<Eigen::half>& output_c_backprop_data, DeviceMemory<Eigen::half>* input_backprop_data, DeviceMemory<Eigen::half>* input_h_backprop_data, DeviceMemory<Eigen::half>* input_c_backprop_data, DeviceMemory<Eigen::half>* params_backprop_data, DeviceMemory<uint8>* reserve_space_data, ScratchAllocator* workspace_allocator) argument
2009 DoRnnBackward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<float>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<float>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, const DeviceMemory<float>& output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, const DeviceMemory<float>& output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, const DeviceMemory<float>& output_c_data, const DeviceMemory<float>& output_backprop_data, const DeviceMemory<float>& output_h_backprop_data, const DeviceMemory<float>& output_c_backprop_data, DeviceMemory<float>* input_backprop_data, DeviceMemory<float>* input_h_backprop_data, DeviceMemory<float>* input_c_backprop_data, DeviceMemory<float>* params_backprop_data, DeviceMemory<uint8>* reserve_space_data, ScratchAllocator* workspace_allocator) argument
2061 DoRnnBackward( Stream* stream, const dnn::RnnDescriptor& rnn_desc, const dnn::RnnSequenceTensorDescriptor& input_desc, const DeviceMemory<double>& input_data, const dnn::RnnStateTensorDescriptor& input_h_desc, const DeviceMemory<double>& input_h_data, const dnn::RnnStateTensorDescriptor& input_c_desc, const DeviceMemory<double>& input_c_data, const DeviceMemory<double>& params, const dnn::RnnSequenceTensorDescriptor& output_desc, const DeviceMemory<double>& output_data, const dnn::RnnStateTensorDescriptor& output_h_desc, const DeviceMemory<double>& output_h_data, const dnn::RnnStateTensorDescriptor& output_c_desc, const DeviceMemory<double>& output_c_data, const DeviceMemory<double>& output_backprop_data, const DeviceMemory<double>& output_h_backprop_data, const DeviceMemory<double>& output_c_backprop_data, DeviceMemory<double>* input_backprop_data, DeviceMemory<double>* input_h_backprop_data, DeviceMemory<double>* input_c_backprop_data, DeviceMemory<double>* params_backprop_data, DeviceMemory<uint8>* reserve_space_data, ScratchAllocator* workspace_allocator) argument
2302 DoConvolveImpl( Stream* stream, const BatchDescriptor& batch_descriptor, const DeviceMemory<T>& input_data, const FilterDescriptor& filter_descriptor, const DeviceMemory<T>& filter_data, const ConvolutionDescriptor& convolution_descriptor, const BatchDescriptor& output_descriptor, DeviceMemory<T>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
2924 DoConvolve( Stream* stream, const BatchDescriptor& batch_descriptor, const DeviceMemory<float>& input_data, const FilterDescriptor& filter_descriptor, const DeviceMemory<float>& filter_data, const ConvolutionDescriptor& convolution_descriptor, const BatchDescriptor& output_descriptor, DeviceMemory<float>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
2940 DoConvolve( Stream* stream, const BatchDescriptor& batch_descriptor, const DeviceMemory<double>& input_data, const FilterDescriptor& filter_descriptor, const DeviceMemory<double>& filter_data, const ConvolutionDescriptor& convolution_descriptor, const BatchDescriptor& output_descriptor, DeviceMemory<double>* output_data) argument
2952 DoConvolve( Stream* stream, const BatchDescriptor& batch_descriptor, const DeviceMemory<Eigen::half>& input_data, const FilterDescriptor& filter_descriptor, const DeviceMemory<Eigen::half>& filter_data, const ConvolutionDescriptor& convolution_descriptor, const BatchDescriptor& output_descriptor, DeviceMemory<Eigen::half>* output_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3107 DoTransformTensor(Stream* stream, const dnn::BatchDescriptor& input_desc, dnn::DataType input_type, const DeviceMemoryBase& input_data, const dnn::BatchDescriptor& output_desc, dnn::DataType output_type, float scale, DeviceMemoryBase* output_data) argument
3384 DoConvolveBackwardFilterImpl( Stream* stream, const dnn::BatchDescriptor& input_descriptor, const DeviceMemory<T>& input_data, const dnn::BatchDescriptor& output_descriptor_in, DeviceMemory<T> backward_output_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const dnn::FilterDescriptor& filter_descriptor, DeviceMemory<T>* backward_filter_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3595 DoConvolveBackwardFilter( Stream* stream, const dnn::BatchDescriptor& input_descriptor, const DeviceMemory<float>& input_data, const dnn::BatchDescriptor& output_descriptor_in, DeviceMemory<float> backward_output_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const dnn::FilterDescriptor& filter_descriptor, DeviceMemory<float>* backward_filter_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3613 DoConvolveBackwardFilter( Stream* stream, const dnn::BatchDescriptor& input_descriptor, const DeviceMemory<Eigen::half>& input_data, const dnn::BatchDescriptor& output_descriptor_in, DeviceMemory<Eigen::half> backward_output_data, const dnn::ConvolutionDescriptor& convolution_descriptor, const dnn::FilterDescriptor& filter_descriptor, DeviceMemory<Eigen::half>* backward_filter_data, ScratchAllocator* scratch_allocator, const dnn::AlgorithmConfig& algorithm_config, dnn::ProfileResult* output_profile_result) argument
3632 DoConvolveBackwardBiasImpl( Stream* stream, const dnn::BatchDescriptor& input_descriptor, const DeviceMemory<T>& input_data, const dnn::BatchDescriptor& bias_descriptor, DeviceMemory<T>* backward_bias_data) argument
3665 DoConvolveBackwardBias( Stream* stream, const BatchDescriptor& input_descriptor, const DeviceMemory<double>& input_data, const BatchDescriptor& bias_descriptor, DeviceMemory<double>* backward_bias_data) argument
3674 DoConvolveBackwardBias( Stream* stream, const BatchDescriptor& input_descriptor, const DeviceMemory<float>& input_data, const BatchDescriptor& bias_descriptor, DeviceMemory<float>* backward_bias_data) argument
3683 DoConvolveBackwardBias( Stream* stream, const BatchDescriptor& input_descriptor, const DeviceMemory<Eigen::half>& input_data, const BatchDescriptor& bias_descriptor, DeviceMemory<Eigen::half>* backward_bias_data) argument
3692 DoMatMul(Stream* stream, const DeviceMemory<float>& input_data, const DeviceMemory<float>& weights, const dnn::BatchDescriptor& input_dimensions, const dnn::BatchDescriptor& output_dimensions, DeviceMemory<float>* output_data) argument
3827 DoBiasAdd(Stream* stream, const DeviceMemory<float>& input_data, const DeviceMemory<float>& biases, const dnn::BatchDescriptor& dimensions, DeviceMemory<float>* output_data) argument
3885 DoActivate(Stream* stream, dnn::ActivationMode activation_mode, const dnn::BatchDescriptor& dimensions, const DeviceMemory<float>& input_data, DeviceMemory<float>* output_data, uint64 options) argument
3954 DoPoolForward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<double>& input_data, const dnn::BatchDescriptor& output_dimensions, DeviceMemory<double>* output_data) argument
3989 DoPoolForward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<float>& input_data, const dnn::BatchDescriptor& output_dimensions, DeviceMemory<float>* output_data) argument
4024 DoPoolForward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<Eigen::half>& input_data, const dnn::BatchDescriptor& output_dimensions, DeviceMemory<Eigen::half>* output_data) argument
4058 DoPoolBackward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<double>& input_data, const dnn::BatchDescriptor& output_dimensions, const DeviceMemory<double>& output_data, const DeviceMemory<double>& input_diff_data, DeviceMemory<double>* output_diff_data) argument
4096 DoPoolBackward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<float>& input_data, const dnn::BatchDescriptor& output_dimensions, const DeviceMemory<float>& output_data, const DeviceMemory<float>& input_diff_data, DeviceMemory<float>* output_diff_data) argument
4134 DoPoolBackward( Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, const dnn::BatchDescriptor& input_dimensions, const DeviceMemory<Eigen::half>& input_data, const dnn::BatchDescriptor& output_dimensions, const DeviceMemory<Eigen::half>& output_data, const DeviceMemory<Eigen::half>& input_diff_data, DeviceMemory<Eigen::half>* output_diff_data) argument
4171 DoNormalize( Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, const DeviceMemory<float>& input_data, DeviceMemory<float>* output_data) argument
4178 DoNormalizeWithDimensions( Stream* stream, const dnn::NormalizeDescriptor& normalize_descriptor, const dnn::BatchDescriptor& dimensions, const DeviceMemory<float>& input_data, DeviceMemory<float>* output_data) argument
4263 DoDepthConcatenate( Stream* stream, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, port::ArraySlice<const DeviceMemory<float>*> input_data, DeviceMemory<float>* output_data) argument
4320 DoElementwiseOperate( Stream* stream, dnn::ElementwiseOperation operation, port::ArraySlice<dnn::BatchDescriptor> input_dimensions, port::ArraySlice<const DeviceMemory<float>*> input_data, const dnn::BatchDescriptor& output_dimensions, DeviceMemory<float>* output_data) argument
4330 DoXYPad(Stream* stream, const dnn::BatchDescriptor& dimensions, const DeviceMemory<float>& input_data, int64 left_pad, int64 right_pad, int64 top_pad, int64 bottom_pad, DeviceMemory<float>* output_data) argument
4339 DoXYSlice(Stream* stream, const dnn::BatchDescriptor& dimensions, const DeviceMemory<float>& input_data, int64 left_trim, int64 right_trim, int64 top_trim, int64 bottom_trim, DeviceMemory<float>* output_data) argument
[all...]

Completed in 477 milliseconds

12345678