| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A D | runtime_fallback_tensor.cc | 67 AbstractTensorInterface* tensor_interface) { in GetShape() argument
69 int64_t num_dims = tensor_interface->NumDims(); in GetShape()
72 dims.push_back(tensor_interface->Dim(i)); in GetShape()
78 AbstractTensorInterface* tensor_interface, HostContext* host) { in CopyTfStringTensorToStringHostTensor() argument
80 TensorMetadata(DType(DType::String), GetShape(tensor_interface)), host); in CopyTfStringTensorToStringHostTensor()
85 assert(tensor_interface->Type() == DT_STRING); in CopyTfStringTensorToStringHostTensor()
86 const int64_t num_elems = tensor_interface->NumElements(); in CopyTfStringTensorToStringHostTensor()
88 reinterpret_cast<const tensorflow::tstring*>(tensor_interface->Data()); in CopyTfStringTensorToStringHostTensor()
102 OwnedAbstractTensorInterface tensor_interface{ in Print() local
106 int rank = tensor_interface->NumDims(); in Print()
[all …]
|
| H A D | conversion_function.cc | 45 OwnedAbstractTensorInterface tensor_interface{ in ConvertRuntimeFallbackTensorToDenseHostTensor() local
51 void* data = tensor_interface->Data(); in ConvertRuntimeFallbackTensorToDenseHostTensor()
52 size_t size = tensor_interface->ByteSize(); in ConvertRuntimeFallbackTensorToDenseHostTensor()
59 [tensor_interface = std::move(tensor_interface)](void*, size_t) {}); in ConvertRuntimeFallbackTensorToDenseHostTensor()
71 OwnedAbstractTensorInterface tensor_interface{ in ConvertRuntimeFallbackTensorToStringHostTensor() local
78 assert(tensor_interface->Type() == DT_STRING); in ConvertRuntimeFallbackTensorToStringHostTensor()
83 CopyTfStringTensorToStringHostTensor(tensor_interface.get(), host_ctx); in ConvertRuntimeFallbackTensorToStringHostTensor()
|
| H A D | runtime_fallback_tensor.h | 60 AbstractTensorInterface* tensor_interface);
63 AbstractTensorInterface* tensor_interface, tfrt::HostContext* host);
|
| H A D | runtime_fallback_kernels.cc | 1223 OwnedAbstractTensorInterface tensor_interface{ in ConvertTFTensorHandleToTFRTTensor() local
1229 auto tf_dtype = tensor_interface->Type(); in ConvertTFTensorHandleToTFRTTensor()
1234 CopyTfStringTensorToStringHostTensor(tensor_interface.get(), host); in ConvertTFTensorHandleToTFRTTensor()
1243 GetShape(tensor_interface.get())); in ConvertTFTensorHandleToTFRTTensor()
1246 void* data = tensor_interface->Data(); in ConvertTFTensorHandleToTFRTTensor()
1247 size_t size = tensor_interface->ByteSize(); in ConvertTFTensorHandleToTFRTTensor()
1254 [tensor_interface = std::move(tensor_interface)](void*, size_t) {}); in ConvertTFTensorHandleToTFRTTensor()
|
| H A D | BUILD | 99 "//tensorflow/c:tensor_interface",
153 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| H A D | BUILD | 28 "//tensorflow/c:tensor_interface",
72 "//tensorflow/c:tensor_interface",
97 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/eigen/bench/btl/libs/tensors/ |
| H A D | main_linear.cpp | 19 bench<Action_axpy<tensor_interface<REAL_TYPE> > >(MIN_AXPY,MAX_AXPY,NB_POINT); in main()
20 bench<Action_axpby<tensor_interface<REAL_TYPE> > >(MIN_AXPY,MAX_AXPY,NB_POINT); in main()
|
| H A D | main_matmat.cpp | 18 bench<Action_matrix_matrix_product<tensor_interface<REAL_TYPE> > >(MIN_MM,MAX_MM,NB_POINT); in main()
|
| H A D | main_vecmat.cpp | 18 bench<Action_matrix_vector_product<tensor_interface<REAL_TYPE> > >(MIN_MV,MAX_MV,NB_POINT); in main()
|
| H A D | tensor_interface.hh | 19 class tensor_interface class
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | BUILD | 38 "tensor_interface.h",
82 "tensor_interface.h",
396 name = "tensor_interface",
397 hdrs = ["tensor_interface.h"],
433 ":tensor_interface",
460 ":tensor_interface",
|
| H A D | c_api.cc | 358 const auto tensor_interface = in TF_TensorToTensorV1() local
369 string(static_cast<const char*>(tensor_interface->Data()), in TF_TensorToTensorV1()
370 tensor_interface->ByteSize()))) { in TF_TensorToTensorV1()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| H A D | BUILD | 126 "//tensorflow/c:tensor_interface",
235 "//tensorflow/c:tensor_interface",
290 "//tensorflow/c:tensor_interface",
|
| H A D | saved_model_utils.cc | 226 TensorInterface tensor_interface(std::move(tensor)); in TensorProtoToConstant() local
227 return Constant::Create(ctx, &tensor_interface, output); in TensorProtoToConstant()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/ci_build/windows/ |
| H A D | libtensorflow_gpu.sh | 73 tensorflow/c/tensor_interface.h \
100 include/tensorflow/c/tensor_interface.h \
|
| H A D | libtensorflow_cpu.sh | 73 tensorflow/c/tensor_interface.h \
100 include/tensorflow/c/tensor_interface.h \
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| H A D | BUILD | 25 "//tensorflow/c:tensor_interface",
44 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | BUILD | 396 "//tensorflow/c:tensor_interface",
424 "//tensorflow/c:tensor_interface",
442 "//tensorflow/c:tensor_interface",
502 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/ |
| H A D | BUILD | 26 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/function/ |
| H A D | BUILD | 75 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | c_api_tfrt.cc | 773 auto* tensor_interface = down_cast<TensorInterface*>(t); in CreateLocalHandle() local
778 if (tensor_interface->IsTfTensor()) { in CreateLocalHandle()
781 tensor_interface->TfTensor())}; in CreateLocalHandle()
803 auto tensor_av = tensor_interface->TensorRef(); in CreateLocalHandle()
|
| H A D | BUILD | 79 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/ |
| H A D | BUILD | 340 "//tensorflow/c:tensor_interface",
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | BUILD | 1134 "tensor_interface.h",
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/libtflite_flex_delegate/android_arm64_armv8-2a_cortex-a55_static_apex33/obj/external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_tensor_handle.o.d | 1218 external/tensorflow/tensorflow/c/tensor_interface.h
|