Home
last modified time | relevance | path

Searched refs:kernel_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
H A Dmorphological_ops_test.py205 kernel_tensor = constant_op.constant(
208 def compute_dilation2d(image_tensor, kernel_tensor): argument
211 kernel_tensor,
222 lambda x: compute_dilation2d(x, kernel_tensor), [image_tensor]))
225 lambda x: compute_dilation2d(image_tensor, x), [kernel_tensor]))
511 kernel_tensor = constant_op.constant(
514 def compute_erosion2d(image_tensor, kernel_tensor): argument
517 kernel_tensor,
528 lambda x: compute_erosion2d(x, kernel_tensor), [image_tensor]))
531 lambda x: compute_erosion2d(image_tensor, x), [kernel_tensor]))
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc102 acl_conv_obj.kernel_tensor.allocator()->init(acl_conf.kernel_info); in ACLDepthwiseConvImpl()
106 &acl_conv_obj.input_tensor, &acl_conv_obj.kernel_tensor, nullptr, in ACLDepthwiseConvImpl()
115 acl_conv_obj.kernel_tensor.allocator()->import_memory(rhs); in ACLDepthwiseConvImpl()
121 acl_conv_obj.kernel_tensor.allocator()->free(); in ACLDepthwiseConvImpl()
206 acl_conv_obj.kernel_tensor.allocator()->init(acl_conf.kernel_info); in ACLGemmConvImpl()
211 &acl_conv_obj.input_tensor, &acl_conv_obj.kernel_tensor, nullptr, in ACLGemmConvImpl()
220 acl_conv_obj.kernel_tensor.allocator()->import_memory(rhs); in ACLGemmConvImpl()
226 acl_conv_obj.kernel_tensor.allocator()->free(); in ACLGemmConvImpl()
H A Druntime_conv2d_acl.h32 arm_compute::Tensor kernel_tensor; member
43 arm_compute::Tensor kernel_tensor; member
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpost_grad.py530 kernel_tensor = ir.TensorBox.create(kernel)
533 dst = ir.SliceView.create(kernel_tensor, dim, offsets_start[i], offsets_end[i])
542 return kernel_tensor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc1596 auto kernel_tensor = op.rhs(); in PrecheckForXlaConvV2Op() local
1606 kernel_tensor.getType().cast<TensorType>().hasStaticShape() && in PrecheckForXlaConvV2Op()
1622 kernel_tensor.getType().cast<RankedTensorType>().getShape(); in PrecheckForXlaConvV2Op()
1729 auto kernel_tensor = op.rhs(); in InferShapeForXlaConvV2Op() local
1756 auto kernel_tensor_shape = kernel_tensor.getType().cast<RankedTensorType>(); in InferShapeForXlaConvV2Op()