| /aosp_15_r20/external/executorch/backends/vulkan/tools/gpuinfo/include/ |
| H A D | textures.h | 61 vTensor in_tensor = in tex_cacheline_concurr() local
161 vTensor in_tensor = api::vTensor(api::context(), sizes_nchw, vkapi::kFloat); in tex_bandwidth() local
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/ |
| H A D | DependencyGraph.h | 194 for(auto in_tensor : inputs) variable
368 void link_input(OperatorId op, TensorId in_tensor) in link_input()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/python/ |
| H A D | toco_from_protos_test.py | 33 def _run(self, sess, in_tensor, out_tensor, should_succeed): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | TensorTransformations.cu | 115 auto in_tensor = self; in roll_cuda() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_transpose_op.cc | 65 Status MKLTransposeND(OpKernelContext* context, const Tensor& in_tensor, in MKLTransposeND()
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _tp_conv.py | 40 def _ring_send_recv_construct(in_tensor, d1, d2, left, right, rank, size): argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Staging.cpp | 67 const ValueRef in_tensor, in add_tensor_to_staging_node()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/python/ |
| H A D | lite_v2_test.py | 3438 def model(in_tensor): argument
3613 def model(in_tensor): argument
3637 def model(in_tensor): argument
|
| H A D | lite_flex_test.py | 401 def conv_func(self, in_tensor, filter_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | collective_ops_test.py | 958 def collective_fn(in_tensor): argument
1008 def collective_fn(in_tensor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | image_grad_test_base.py | 144 def func(in_tensor): argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/ |
| H A D | VulkanBackend.cpp | 394 vTensorPtr in_tensor = graph->get_tensor(in_tensor_ref); in maybe_resize_input() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | interpreter_test.cc | 1479 const TfLiteTensor* in_tensor; in CancelOpRegistration() local
1501 const TfLiteTensor* in_tensor; in OkOpRegistration() local
|