Home
last modified time | relevance | path

Searched refs:input_channels (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dquantized_pooling_ops_test.cc55 const int input_channels = 2; in TEST_F() local
56 Tensor input_float(DT_FLOAT, {1, input_height, input_width, input_channels}); in TEST_F()
67 {1, expected_height, expected_width, input_channels}); in TEST_F()
100 const int input_channels = 2; in TEST_F() local
101 Tensor input_float(DT_FLOAT, {1, input_height, input_width, input_channels}); in TEST_F()
112 {1, expected_height, expected_width, input_channels}); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc28 int64 input_channels, int64 kernel_rows, int64 kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc29 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenConvF32() argument
39 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF32()
48 int64 input_channels, int64 kernel_rows, int64 kernel_cols, in __xla_cpu_runtime_EigenConvF16() argument
58 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d.h27 tensorflow::int64 input_channels, tensorflow::int64 kernel_rows,
41 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
Druntime_single_threaded_conv2d.h27 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
41 tensorflow::int64 input_channels, tensorflow::int64 kernel_rows,
Druntime_conv2d_impl.h30 int64 input_cols, int64 input_channels, int64 kernel_rows, in EigenConvImpl() argument
40 input(lhs, input_batch, input_rows, input_cols, input_channels); in EigenConvImpl()
Dir_emitter.cc886 int64 input_channels = in HandleConvolution() local
970 ir_builder_.getInt64(input_channels), in HandleConvolution()
/external/adhd/cras/src/server/
Dcras_dsp_pipeline.c126 int input_channels; member
452 pipeline->input_channels = ARRAY_COUNT( in cras_dsp_pipeline_create()
456 if (pipeline->output_channels > pipeline->input_channels) { in cras_dsp_pipeline_create()
742 return pipeline->input_channels; in cras_dsp_pipeline_get_num_input_channels()
824 unsigned int input_channels = pipeline->input_channels; in cras_dsp_pipeline_apply() local
826 float *source[input_channels]; in cras_dsp_pipeline_apply()
838 for (i = 0; i < input_channels; i++) in cras_dsp_pipeline_apply()
850 dsp_util_deinterleave(target, source, input_channels, chunk); in cras_dsp_pipeline_apply()
942 dumpf(d, " input channels: %d\n", pipeline->input_channels); in cras_dsp_pipeline_dump()
/external/webrtc/webrtc/common_audio/include/
Daudio_util.h137 void DownmixToMono(const T* const* input_channels, in DownmixToMono() argument
142 Intermediate value = input_channels[0][i]; in DownmixToMono()
144 value += input_channels[j][i]; in DownmixToMono()
/external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
Dfisher_blocks_test.py556 self.input_channels = 6
561 self.input_channels).astype(np.float32)
569 self.input_channels, self.output_channels).astype(
604 batch_size, height, width, input_channels = inputs.shape
615 inputs = np.reshape(inputs, [batch_size, num_locations, input_channels])
619 fisher_diag = np.zeros((input_channels, output_channels))
622 example_fisher_diag = np.zeros((input_channels, output_channels))
637 self.kernel_size, self.kernel_size, self.input_channels,
650 self.kernel_size, self.kernel_size, self.input_channels,
681 self.kernel_size, self.kernel_size, self.input_channels + 1,
/external/webrtc/webrtc/modules/audio_processing/test/
Ddebug_dump_test.cc45 int input_channels,
115 int input_channels, in DebugDumpGenerator() argument
121 : input_config_(input_rate_hz, input_channels), in DebugDumpGenerator()
123 output_config_(input_rate_hz, input_channels), in DebugDumpGenerator()
125 input_file_channels_(input_channels), in DebugDumpGenerator()
/external/tensorflow/tensorflow/core/ops/
Daudio_ops.cc79 DimensionHandle input_channels = c->Dim(input, 1); in SpectrogramShapeFn() local
99 c->MakeShape({input_channels, output_length, output_channels})); in SpectrogramShapeFn()
/external/tensorflow/tensorflow/core/util/
Dtensor_format.h506 const int64 input_channels = in ShapeFromFilterFormat() local
516 input_channels, output_channels); in ShapeFromFilterFormat()
523 input_channels, output_channels); in ShapeFromFilterFormat()
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/
Dfused_conv2d_bias_activation_op_test.py800 input_channels = test_param["input_channels"]
815 [batch_size, input_channels // 4, input_height, input_width, 4],
823 output_channels, input_channels // 4, filter_height,
/external/tensorflow/tensorflow/python/layers/
Dnormalization_test.py106 batch, height, width, input_channels = 2, 4, 5, 3
107 shape = [batch, height, width, input_channels]
139 image_val = np.random.rand(batch, height, width, input_channels).astype(
168 batch, height, width, input_channels = 2, 4, 5, 3
169 shape = [batch, height, width, input_channels]
215 input_channels).astype(np.float32)
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc135 int input_channels = 3; in SimpleFusedBatchNormGrad() local
136 TensorShape shape({batch_size, input_height, input_width, input_channels}); in SimpleFusedBatchNormGrad()
143 TensorShape shape_vector({input_channels}); in SimpleFusedBatchNormGrad()
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc1944 const int64 input_channels = in HandleConvolution() local
1960 input_shape.element_type(), {conv_width, input_channels}); in HandleConvolution()
1965 filter_shape.element_type(), {input_channels, output_channels}); in HandleConvolution()
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py226 input_channels = 3
227 shape = [batch, height, width, input_channels]