Home
last modified time | relevance | path

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

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