Home
last modified time | relevance | path

Searched defs:tf_tensor (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/
H A Dconversion.cc63 const auto* tf_tensor = dst_knfb_tensor->GetTensor(); in ConvertKernelFallbackTensorToStringHostTensor() local
86 auto tf_tensor = CopyShtToTfTensor(tensor); in ConvertStringHostTensorToKernelFallbackTensor() local
106 const auto* tf_tensor = dst_knfb_tensor->GetTensor(); in ConvertKernelFallbackTensorToDenseHostTensor() local
122 auto tf_tensor = in ConvertDenseHostTensorToKernelFallbackTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dtflite_subgraph_execute.cc115 const Tensor& tf_tensor = ctx->input(i + 1); in ResizeInputTensor() local
166 const Tensor& tf_tensor = ctx->input(i + 1); in SetSubgraphInput()
H A Dutil_test.cc220 tensorflow::Tensor tf_tensor = tf_tensor_or.ValueOrDie(); in TEST() local
258 tensorflow::Tensor tf_tensor = tf_tensor_or.ValueOrDie(); in TEST() local
H A Dkernel.cc318 tensorflow::Tensor* tf_tensor, in CopyToTfLiteTensor()
352 tensorflow::Tensor& tf_tensor = tensors->at(i); in MaybePersistTensorflowOutputs() local
778 const tensorflow::Tensor& tf_tensor = buffer_map->GetTensor(tensor_index); in Eval() local
H A Dbuffer_map_util.cc135 tensorflow::Tensor* tf_tensor, in SetTfTensorFromTfLite()
H A Dutil.cc230 tensorflow::Tensor tf_tensor( in CreateTfTensorFromTfLiteTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc333 OwnedTFTensor tf_tensor{ in MoveDHTToTFTensor() local
349 OwnedTFTensor tf_tensor = MoveDHTToTFTensor(std::move(*dht), host); in DecodeDenseAttrToTensorInterface() local
827 llvm::Expected<tensorflow::Tensor> tf_tensor) { in CoreRTTensorHandleToFallbackTensorInternal()
1006 auto& tf_tensor = av->get<tensorflow::tfrt_stub::FallbackTensor>().tensor(); in FallbackTensorToCoreRTTensorHandleInternal() local
1129 auto tf_tensor = av->get<tensorflow::Tensor>(); in RuntimeFallbackExecuteOp() local
1166 const tensorflow::Tensor* tf_tensor = nullptr; in RuntimeFallbackExecuteOp() local
H A Druntime_fallback_batch_tf_opkernels.cc73 const tensorflow::Tensor* tf_tensor; in ConvertTFRTTensorToTFTensor() local
482 const tensorflow::Tensor& tf_tensor) { in TFTensorToFallbackTensor()
/aosp_15_r20/external/tensorflow/tensorflow/c/kernels/
H A Dtensor_shape_utils_test.cc32 TF_Tensor* tf_tensor; member
40 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status); in TestShapeMatch() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcore.cc57 TF_Tensor* tf_tensor = TF_TensorFromTensor(*t, status); in Resolve() local
95 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, status); in Resolve() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Deval_util.cc109 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status->status); in EvaluateOperation() local
135 TF_Tensor* tf_tensor = TFE_TensorHandleResolve(tensor_handle, status); in EvaluateOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtf_op_shim.cc86 const auto tf_tensor = context_->input(idx); in GetInput() local
142 const auto* tf_tensor = context_->input_tensor(idx); in GetInputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dattr_util.cc243 llvm::Expected<tensorflow::Tensor> tf_tensor = in FillAttrValueMapUsingScalar() local
557 auto tf_tensor = CreateTfTensorFromDenseAttr(dense_attr); in SetUpScalarAttr() local
597 auto tf_tensor = CreateTfTensorFromDenseAttr(dense_attr); in AddTensorToAttrList() local
H A Dtensor_util.h58 const tensorflow::Tensor& tf_tensor) { in GetTensorMetadata()
H A Dtensor_util.cc46 OwnedTFTensor tf_tensor{TF_NewTensor( in MoveHostBufferToTfTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtensor_util.cc68 tensorflow::Tensor tf_tensor( in CopyScalarHostTensorToTFTensor() local
135 const tensorflow::Tensor& tf_tensor, HostContext* host_ctx) { in TFTensorToTFRTTensorHandle()
/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dndarray_tensor.cc559 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NdarrayToTensor() local
569 Safe_TF_TensorPtr tf_tensor = make_safe(TF_TensorFromTensor(t, &status)); in TensorToNdarray() local
H A Dpy_seq_tensor.cc689 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NumpyToTFE_TensorHandle() local
/aosp_15_r20/external/tensorflow/tensorflow/python/dlpack/
H A Ddlpack.py23 def to_dlpack(tf_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/conversion/
H A Dconversion.cc66 const tensorflow::Tensor *tf_tensor; in ConvertRuntimeFallbackToKernelFallbackTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_experimental.cc102 TF_Tensor* tf_tensor = TF_TensorFromTensor(*var->tensor(), &s); in EnsureSparseVariableAccess() local
139 TF_Tensor* tf_tensor = TF_TensorFromTensor(*tensor, &s); in PrepareToUpdateVariable() local
H A Dkernels.cc698 TF_Tensor* tf_tensor = TF_TensorFromTensor(*tensor, &s); in TF_AllocateOutput() local
767 TF_Tensor* tf_tensor; in TF_AllocateTemp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_input_conversion_op.cc181 const Tensor* tf_tensor; in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/
H A Dconversion_function.cc112 const tensorflow::Tensor* tf_tensor = nullptr; in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/teflon/
H A Dtfl_device.c133 TfLiteTensor tf_tensor = tf_context->tensors[index]; in fill_tensor() local

12