Searched defs:input_c (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 571 const Tensor** input_c, const Tensor** params, in ExtractForwardInput()
685 const Tensor** input_c, const Tensor** params, in ExtractForwardInput()
794 const Tensor* input_c, const Tensor* params, in DoForward()
873 const Tensor* input, const Tensor* input_h, const Tensor* input_c, in DoBackward()
1502 const Tensor* input_c = nullptr; in ComputeAndReturnAlgorithm() local
1572 const Tensor* input_c, const Tensor* params, in MaybeAutotune()
1678 const Tensor* input_h, const Tensor* input_c, in MaybeAutotune()
1904 const Tensor* input_c = nullptr; in ComputeImpl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/ |
| H A D | xla_compilation_cache_test_helper.cc | 97 Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements()); in ExecuteWithBatch() local
120 Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements()); in ExecuteWithBatch() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuIm2ColKernel.cpp | 190 int input_c, in linearize_volume_nhwc()
263 const int input_c = src->info()->dimension(channel_idx); in run_im2col() local
|
| /aosp_15_r20/external/libaom/test/ |
| H A D | fft_test.cc | 212 std::vector<std::complex<float> > input_c = fft2d(&expected[0], n); in TEST_P() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/xpu/ |
| H A D | Conv.cpp | 582 Tensor input_c = input_r.contiguous(backend_memory_format); in convolution_overrideable() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Normalization.cpp | 587 auto input_c = input.contiguous(input.suggest_memory_format()); in _batch_norm_impl_index() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 519 for (int input_c = begin_input_c; input_c < end_input_c; ++input_c) { in LocalResponseNormalization() local
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/ |
| H A D | vulkan_compute_api_test.cpp | 1818 int64_t input_c = 8; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/ |
| H A D | pass.cc | 2540 Operation *input_c = const_op; in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding.cc | 3430 const string input_c = in ConstantPushDown() local
|