Home
last modified time | relevance | path

Searched defs:result_tensor (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_unified_experimental_test.cc96 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
161 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
244 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
330 TF_Tensor* result_tensor = TFE_TensorHandleResolve(result_t, status.get()); in TEST_P() local
H A Dunified_api_testutil.cc133 Status GetValue(AbstractTensorHandle* t, TF_Tensor** result_tensor) { in GetValue()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dhashtable_ops_test.cc288 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetLookupResult() local
301 auto* result_tensor = interpreter_->tensor(kResultTwoTensorId); in GetLookupTwoResult() local
314 auto* result_tensor = interpreter_->tensor(kResultTensorId); in GetStringLookupResult() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_tensor_coding_test.cc48 Tensor result_tensor; in Validate() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/
H A Dget_options_op_test.cc91 Tensor result_tensor = output[0]; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dcustom_gradient_test.cc118 TF_Tensor* result_tensor; in TEST_P() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DLegacyBatching.cpp75 auto result_tensor = makeBatched(batched->value(), std::move(new_bdims)); in remove_existing_batch_dim() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_op_handler.cc120 const RuntimeFallbackTensor& result_tensor, in GetDeviceFromFallbackTensor()
/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupGlooAsyncTest.cpp209 auto& result_tensor = results[j]; in runAsyncAllreduceTest() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dcwise_ops_unary_test.py117 def _check(self, result_tensor, result_np, input_sp_t, tol): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mlir_generated/
H A Dbase_op.h120 Tensor result_tensor = ConvertDescriptorToTensor<OutputDataType>( in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_ops_test.py884 def _check(self, result_tensor, result_np, input_sp_t): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/client/
H A Dtf_session_helper.cc704 TF_Tensor* result_tensor; in TF_TryEvaluateConstant_wrapper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.cc826 auto result_tensor = MakeIndirectAsyncValue(host); in CreateLocalHandle() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api.cc2424 Tensor result_tensor; in TF_TryEvaluateConstant() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc4387 ITensorProxyPtr result_tensor = nullptr; in ConvertGather() local