| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/conversion/ |
| D | conversion.cc | 63 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
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | tflite_subgraph_execute.cc | 115 const Tensor& tf_tensor = ctx->input(i + 1); in ResizeInputTensor() local
166 const Tensor& tf_tensor = ctx->input(i + 1); in SetSubgraphInput()
|
| D | util_test.cc | 220 tensorflow::Tensor tf_tensor = tf_tensor_or.ValueOrDie(); in TEST() local
258 tensorflow::Tensor tf_tensor = tf_tensor_or.ValueOrDie(); in TEST() local
|
| D | buffer_map_util.cc | 135 tensorflow::Tensor* tf_tensor, in SetTfTensorFromTfLite()
|
| D | util.cc | 230 tensorflow::Tensor tf_tensor( in CreateTfTensorFromTfLiteTensor() local
|
| D | kernel.cc | 318 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
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | tensor_shape_utils_test.cc | 32 TF_Tensor* tf_tensor; member
40 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, &status); in TestShapeMatch() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | core.cc | 57 TF_Tensor* tf_tensor = TF_TensorFromTensor(*t, status); in Resolve() local
95 TF_Tensor* tf_tensor = TF_TensorFromTensor(tensor, status); in Resolve() local
|
| /external/tensorflow/tensorflow/python/dlpack/ |
| D | dlpack.py | 23 def to_dlpack(tf_tensor): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | eval_util.cc | 109 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
|
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | tf_op_shim.cc | 86 const auto tf_tensor = context_->input(idx); in GetInput() local
142 const auto* tf_tensor = context_->input_tensor(idx); in GetInputTensor() local
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | tensor_util.cc | 68 tensorflow::Tensor tf_tensor( in CopyScalarHostTensorToTFTensor() local
135 const tensorflow::Tensor& tf_tensor, HostContext* host_ctx) { in TFTensorToTFRTTensorHandle()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | tensor_util.h | 58 const tensorflow::Tensor& tf_tensor) { in GetTensorMetadata()
|
| D | tensor_util.cc | 46 OwnedTFTensor tf_tensor{TF_NewTensor( in MoveHostBufferToTfTensor() local
|
| D | attr_util.cc | 243 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
|
| /external/tensorflow/tensorflow/core/runtime_fallback/conversion/ |
| D | conversion.cc | 66 const tensorflow::Tensor *tf_tensor; in ConvertRuntimeFallbackToKernelFallbackTensor() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 333 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
|
| D | runtime_fallback_batch_tf_opkernels.cc | 73 const tensorflow::Tensor* tf_tensor; in ConvertTFRTTensorToTFTensor() local
482 const tensorflow::Tensor& tf_tensor) { in TFTensorToFallbackTensor()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor.cc | 559 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
|
| D | py_seq_tensor.cc | 689 Safe_TF_TensorPtr tf_tensor = make_safe(static_cast<TF_Tensor*>(nullptr)); in NumpyToTFE_TensorHandle() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 181 const Tensor* tf_tensor; in Compute() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/gpu/ |
| D | conversion_function.cc | 112 const tensorflow::Tensor* tf_tensor = nullptr; in ConvertRuntimeFallbackTensorToDenseGpuTensor() local
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_experimental.cc | 102 TF_Tensor* tf_tensor = TF_TensorFromTensor(*var->tensor(), &s); in EnsureSparseVariableAccess() local
139 TF_Tensor* tf_tensor = TF_TensorFromTensor(*tensor, &s); in PrepareToUpdateVariable() local
|
| D | kernels.cc | 698 TF_Tensor* tf_tensor = TF_TensorFromTensor(*tensor, &s); in TF_AllocateOutput() local
767 TF_Tensor* tf_tensor; in TF_AllocateTemp() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device_util.cc | 101 std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tf_tensor( in BroadcastResourceTensor() local
|