| /external/pytorch/torchgen/ |
| D | gen_executorch.py | 293 kernel_index: ETKernelIndex,
305 for kernel_key, metadata in kernel_index.get_kernels(native_function).items()
333 g: NativeFunctionsGroup | NativeFunction, kernel_index: ETKernelIndex
337 metadata_list = kernel_index.get_kernels(g).values()
356 kernel_index: ETKernelIndex,
374 backend_index = kernel_index._to_backend_index()
410 kernel_index: ETKernelIndex,
422 op_kernels = kernel_index.get_kernels(f)
433 native_function_decl_gen(f, kernel_index)
444 kernel_index: ETKernelIndex,
[all …]
|
| /external/vboot_reference/scripts/image_signing/ |
| D | make_dev_ssd.sh | 221 for kernel_index in $FLAGS_partitions; do
224 local name="$(cros_kernel_name $kernel_index)"
225 local rootfs_index="$(($kernel_index + 1))"
229 offset="$(partoffset "$ssd_device" "$kernel_index")" ||
230 die "Failed to get partition ${kernel_index} offset from ${ssd_device}"
231 size="$(partsize "$ssd_device" "$kernel_index")" ||
232 die "Failed to get partition ${kernel_index} size from ${ssd_device}"
238 debug_msg "Reading $name from partition $kernel_index"
252 old_config_file="${FLAGS_save_config}.$kernel_index"
262 new_config_file="${FLAGS_set_config}.$kernel_index"
[all …]
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolutionSycl.h | 254 const size_t kernel_index = i + kernel_size[0] * (j + kernel_size[1] * k);
259 result += (local_acc[local_index] * kernel_ptr[kernel_index]);
417 …auto kernel_index = std::array<size_t, 2>{static_cast<int>(Layout) == static_cast<int>(ColMajor) ?…
419 auto kernel_size = cl::sycl::range<2>{(size_t)m_kernelImpl.dimensions()[kernel_index[0]],
420 (size_t)m_kernelImpl.dimensions()[kernel_index[1]]};
421 const size_t numX = dimensions()[m_indices[kernel_index[0]]];
422 const size_t numY = dimensions()[m_indices[kernel_index[1]]];
434 const array<Index, 2> indices{{m_indices[kernel_index[0]], m_indices[kernel_index[1]]}};
436 … {m_kernelImpl.dimensions()[kernel_index[0]], m_kernelImpl.dimensions()[kernel_index[1]]}};
448 …auto kernel_index = std::array<size_t, 3>{static_cast<int>(Layout) == static_cast<int>(ColMajor) ?…
[all …]
|
| /external/libgav1/src/dsp/arm/ |
| D | intra_edge_neon.cc | 45 const int kernel_index = strength - 1; in IntraEdgeFilter_NEON() local
60 kKernelsNEON[kernel_index][0] * dst_buffer[size - 2] + in IntraEdgeFilter_NEON()
61 kKernelsNEON[kernel_index][1] * dst_buffer[size - 1] + in IntraEdgeFilter_NEON()
62 kKernelsNEON[kernel_index][0] * dst_buffer[size - 1], in IntraEdgeFilter_NEON()
65 const uint8x8_t krn1 = vdup_n_u8(kKernelsNEON[kernel_index][1]); in IntraEdgeFilter_NEON()
79 sum_lo = vmulq_n_u16(sum_lo, kKernelsNEON[kernel_index][0]); in IntraEdgeFilter_NEON()
82 sum_hi = vmulq_n_u16(sum_hi, kKernelsNEON[kernel_index][0]); in IntraEdgeFilter_NEON()
103 sum_lo = vmulq_n_u16(sum_lo, kKernelsNEON[kernel_index][0]); in IntraEdgeFilter_NEON()
106 sum_hi = vmulq_n_u16(sum_hi, kKernelsNEON[kernel_index][0]); in IntraEdgeFilter_NEON()
317 const int kernel_index = strength - 1; in IntraEdgeFilter_NEON() local
[all …]
|
| /external/pytorch/torchgen/executorch/ |
| D | model.py | 162 kernel_index: dict[OperatorName, dict[ETKernelKey, BackendMetadata]],
168 if op in kernel_index:
169 kernel_index[op][ETKernelKey(default=True)] = backend_metadata
171 kernel_index[op] = {ETKernelKey(default=True): backend_metadata}
177 kernel_index: dict[
180 ETKernelIndex.grow_from_backend_indices(kernel_index, backend_indices)
181 return ETKernelIndex(kernel_index)
|
| /external/pytorch/tools/test/ |
| D | test_executorch_gen.py | 212 op_entries = parsed_yaml.kernel_index.index
298 op_entries = parsed_yaml.kernel_index.index
365 self.kernel_index = ETKernelIndex.from_backend_indices(backend_indices)
373 kernel_index=self.kernel_index,
410 kernel_index=self.kernel_index,
433 kernel_index=self.kernel_index,
|
| D | test_executorch_custom_ops.py | 123 kernel_index=ETKernelIndex(index={}),
143 kernel_index=ETKernelIndex(index={}),
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | weights_conversion.h | 345 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialI4O4() local
346 const int kernel_index_x = kernel_index % weights.shape.w; in RearrangeWeightsToOICustomSpatialI4O4()
347 const int kernel_index_y = kernel_index / weights.shape.w; in RearrangeWeightsToOICustomSpatialI4O4()
382 int kernel_index = in RearrangeWeightsToOICustomSpatialI4O4() local
384 const int kernel_index_x = kernel_index % weights.shape.w; in RearrangeWeightsToOICustomSpatialI4O4()
385 kernel_index /= weights.shape.w; in RearrangeWeightsToOICustomSpatialI4O4()
386 const int kernel_index_y = kernel_index % weights.shape.h; in RearrangeWeightsToOICustomSpatialI4O4()
387 const int kernel_index_z = kernel_index / weights.shape.h; in RearrangeWeightsToOICustomSpatialI4O4()
423 const int kernel_index = spatial_remap[y * weights.shape.w + x]; in RearrangeWeightsToOICustomSpatialO4I4() local
424 const int kernel_index_x = kernel_index % weights.shape.w; in RearrangeWeightsToOICustomSpatialO4I4()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 135 const size_t kernel_index = const_tensor_kernels_.size(); in Initialize() local
139 const_tensor_kernels_[kernel_index]; in Initialize()
143 const size_t kernel_index = kernels_.size(); in Initialize() local
146 KernelState& kernel_state = kernels_[kernel_index]; in Initialize()
150 node_to_index_map[n] = kernel_index; in Initialize()
151 if (kernel_index == 0) { in Initialize()
154 const KernelState& previous_kernel_state = kernels_[kernel_index - 1]; in Initialize()
|
| /external/libgav1/src/dsp/ |
| D | intra_edge.cc | 39 const int kernel_index = strength - 1; in IntraEdgeFilter_C() local
44 sum += kKernels[kernel_index][j] * edge[k]; in IntraEdgeFilter_C()
|
| /external/pytorch/torchgen/executorch/api/ |
| D | custom_ops.py | 87 kernel_index: ETKernelIndex,
104 backend_index = kernel_index._to_backend_index()
|
| /external/libgav1/src/dsp/x86/ |
| D | intra_edge_sse4.cc | 193 const int kernel_index = strength - 1; in IntraEdgeFilter_SSE4_1() local
199 sum += kKernels[kernel_index][j] * edge[k]; in IntraEdgeFilter_SSE4_1()
|
| /external/XNNPACK/src/ |
| D | indirection.c | 60 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_conv2d() local
61 …const size_t index = output_tile_start * kernel_size + kernel_index * output_tile_size + output_ti… in xnn_indirection_init_conv2d()
71 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_conv2d() local
72 …const size_t index = output_tile_start * kernel_size + kernel_index * output_tile_size + output_ti… in xnn_indirection_init_conv2d()
123 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_deconv2d() local
124 …const size_t index = output_tile_start * kernel_size + kernel_index * output_tile_size + output_ti… in xnn_indirection_init_deconv2d()
|
| /external/executorch/runtime/executor/ |
| D | method.h | 344 size_t kernel_index,
|
| D | method.cpp | 495 size_t kernel_index, in resolve_operator() argument
557 kernels[kernel_index] = op_function.get(); in resolve_operator()
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | r600_pipe.h | 185 unsigned kernel_index; member
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | elemental_ir_emitter.cc | 3084 llvm::Value* kernel_index, in EmitConvolution()
3089 NSWMul(kernel_index, b_->getInt64(window_dim.window_dilation())); in EmitConvolution()
3164 llvm_ir::IrArray::Index kernel_index(kernel_multi_index, rhs->shape(), in EmitConvolution() local
3167 kernel_generator(kernel_index)); in EmitConvolution()
|