Home
last modified time | relevance | path

Searched defs:handle_tensor (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dxrt_state_ops.h377 const Tensor& handle_tensor = ctx->input(0); in Compute() local
500 const Tensor& handle_tensor = ctx->input(0); in Compute() local
551 const Tensor& handle_tensor = ctx->input(0); in Compute() local
635 const Tensor& handle_tensor = ctx->input(0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/simple_hash_table/
H A Dsimple_hash_table_kernel.cc140 Tensor handle_tensor; in Compute() local
162 const Tensor& handle_tensor = ctx->input(0); in GetResource() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Ddebug.py638 def handle_tensor(x: Any) -> Any: function
679 def handle_tensor(x: Any) -> Any: function
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dresource_ops_test.cc59 Tensor handle_tensor; in Compute() local
H A Dlookup_util.cc291 const Tensor* handle_tensor; in GetResourceLookupTable() local
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Doutput_graph.py688 def handle_tensor(t, src): function