Home
last modified time | relevance | path

Searched refs:input_h (Results 1 – 25 of 31) sorted by relevance

12

/external/armnn/third-party/stb/
Dstb_image_resize.h225 STBIRDEF int stbir_resize_uint8( const unsigned char *input_pixels , int input_w , int input_h
229 STBIRDEF int stbir_resize_float( const float *input_pixels , int input_w , int input_h , int in…
250 STBIRDEF int stbir_resize_uint8_srgb(const unsigned char *input_pixels , int input_w , int input_h
264 …int8_srgb_edgemode(const unsigned char *input_pixels , int input_w , int input_h , int input_strid…
304 …ize_uint8_generic( const unsigned char *input_pixels , int input_w , int input_h , int input_strid…
310 …ize_uint16_generic(const stbir_uint16 *input_pixels , int input_w , int input_h , int input_strid…
316 …ize_float_generic( const float *input_pixels , int input_w , int input_h , int input_strid…
347 STBIRDEF int stbir_resize( const void *input_pixels , int input_w , int input_h , int input…
355 STBIRDEF int stbir_resize_subpixel(const void *input_pixels , int input_w , int input_h , int input…
365 STBIRDEF int stbir_resize_region( const void *input_pixels , int input_w , int input_h , int input…
[all …]
/external/ComputeLibrary/src/core/NEON/kernels/
DNEIm2ColKernel.cpp105 int input_h, in linearize_volume_nchw() argument
126 if((y < 0 || y >= input_h) && has_pads) in linearize_volume_nchw()
163 if((y < 0 || y >= input_h) && has_pads) in linearize_volume_nchw()
202 int input_h, in linearize_volume_nhwc() argument
214 …if((start_y >= 0) && (end_y < input_h) && (start_x >= 0) && (end_x < input_w) && (dilation_x == 1)… in linearize_volume_nhwc()
227 if(y < 0 || y >= input_h) in linearize_volume_nhwc()
275 const int input_h = _input->info()->dimension(height_idx); in run_im2col() local
317 input_h, in run_im2col()
335 input_h, in run_im2col()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc569 const Tensor** input, const Tensor** input_h, in ExtractForwardInput() argument
574 TF_RETURN_IF_ERROR(context->input("input_h", input_h)); in ExtractForwardInput()
597 if ((*input_h)->dims() != 3) { in ExtractForwardInput()
602 (*input_h)->dim_size(0) / model_shapes->dir_count; in ExtractForwardInput()
605 (*input_h)->dim_size(1) / model_shapes->dir_count; in ExtractForwardInput()
607 model_shapes->num_units = (*input_h)->dim_size(2); in ExtractForwardInput()
619 if ((*input_h)->shape() != model_shapes->hidden_state_shape) { in ExtractForwardInput()
621 "Invalid input_h shape: ", (*input_h)->shape().DebugString(), " ", in ExtractForwardInput()
637 if ((*input_h)->shape() != (*input_c)->shape()) { in ExtractForwardInput()
640 (*input_h)->shape().DebugString(), " ", in ExtractForwardInput()
[all …]
Dconv_grad_input_ops.h120 auto input_h = GetTensorDimIndex(data_format, 'H'); in operator() local
122 strides[input_h] = row_stride; in operator()
124 dilations[input_h] = row_dilation; in operator()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CudnnRNNBackprop.pbtxt18 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
28 output_h: The same shape has input_h.
38 input_h_backprop: The backprop to input_h in the forward pass. Has the same
39 shape as input_h.
Dapi_def_CudnnRNNBackpropV2.pbtxt21 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
31 output_h: The same shape has input_h.
42 input_h_backprop: The backprop to input_h in the forward pass. Has the same
43 shape as input_h.
Dapi_def_CudnnRNNBackpropV3.pbtxt22 input_h: If time_major is true, this is a 3-D tensor with the shape of
35 output_h: The same shape has input_h.
47 input_h_backprop: The backprop to input_h in the forward pass. Has the same
48 shape as input_h.
Dapi_def_CudnnRNN.pbtxt19 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
29 output_h: The same shape has input_h.
Dapi_def_CudnnRNNV2.pbtxt20 input_h: A 3-D tensor with the shape of [num_layer * dir, batch_size,
30 output_h: The same shape has input_h.
Dapi_def_CudnnRNNV3.pbtxt22 input_h: If time_major is true, this is a 3-D tensor with the shape of
35 output_h: The same shape has input_h.
/external/tensorflow/tensorflow/python/ops/
Dcudnn_rnn_grad.py32 input_h=op.inputs[1],
57 input_h=op.inputs[1],
84 input_h=op.inputs[1],
/external/tensorflow/tensorflow/python/keras/layers/
Dcudnn_recurrent.py271 input_h = initial_state[0]
272 input_h = array_ops.expand_dims(input_h, axis=0)
295 'input_h': input_h,
467 input_h = initial_state[0]
469 input_h = array_ops.expand_dims(input_h, axis=0)
497 'input_h': input_h,
Drecurrent_v2.py675 input_h=init_h,
691 input=inputs, input_h=init_h, input_c=0, params=params,
1493 input_h=init_h,
1512 input=inputs, input_h=init_h, input_c=init_c, params=params,
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DCudnnRNNV3.pbtxt8 name: "input_h"
133 name: "input_h"
265 name: "input_h"
DCudnnRNNBackpropV3.pbtxt8 name: "input_h"
154 name: "input_h"
307 name: "input_h"
DCudnnRNN.pbtxt8 name: "input_h"
DCudnnRNNV2.pbtxt8 name: "input_h"
DCudnnRNNBackprop.pbtxt8 name: "input_h"
DCudnnRNNBackpropV2.pbtxt8 name: "input_h"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DCudnnRNNV3.pbtxt8 name: "input_h"
133 name: "input_h"
DCudnnRNNBackpropV3.pbtxt8 name: "input_h"
154 name: "input_h"
DCudnnRNN.pbtxt8 name: "input_h"
DCudnnRNNV2.pbtxt8 name: "input_h"
DCudnnRNNBackpropV2.pbtxt8 name: "input_h"
DCudnnRNNBackprop.pbtxt8 name: "input_h"

12