| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| D | fallback_tensor.h | 50 class FallbackTensor {
52 FallbackTensor() = default;
54 explicit FallbackTensor(const tensorflow::Tensor& tensor) : tensor_(tensor) {} in FallbackTensor() function
55 explicit FallbackTensor(tensorflow::Tensor&& tensor) in FallbackTensor() function
58 explicit FallbackTensor(ImmutableTensor* immutable_tensor) in FallbackTensor() function
70 return const_cast<FallbackTensor*>(this)->tensor(); in tensor()
|
| D | fallback_tensor_test.cc | 52 TEST(FallbackTensorTest, FallbackTensor) { in TEST() argument
57 FallbackTensor fallback_tensor(tensor); in TEST()
69 FallbackTensor fallback_tensor(&immutable_tensor); in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_kernels.cc | 90 using ::tensorflow::tfrt_stub::FallbackTensor;
236 RepeatedArguments<FallbackTensor> operands, in ConvertTensorOperandsToMemrefDesc()
266 RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
311 RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
352 RepeatedArguments<FallbackTensor> operands(o.values()); in ExecuteImpl()
370 static void ExecuteImpl(RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
409 RepeatedArguments<FallbackTensor> operands(o.values()); in ExecuteImpl()
427 static void Execute(RepeatedArguments<FallbackTensor> operands, in Execute()
438 static void ExecuteDebug(RepeatedArguments<FallbackTensor> operands, in ExecuteDebug()
|
| D | tf_jitrt_kernels.cc | 126 using ::tensorflow::tfrt_stub::FallbackTensor;
605 RepeatedArguments<FallbackTensor> operands) { in ConvertTensorOperandsToMemrefDesc()
608 for (FallbackTensor& operand : operands) in ConvertTensorOperandsToMemrefDesc()
649 RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
704 RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
749 RepeatedArguments<FallbackTensor> operands(o.values()); in ExecuteImpl()
766 RepeatedArguments<FallbackTensor> operands) { in OperandsToString()
781 static void ExecuteImpl(RepeatedArguments<FallbackTensor> operands, in ExecuteImpl()
822 RepeatedArguments<FallbackTensor> operands(o.values()); in ExecuteImpl()
839 RepeatedArguments<FallbackTensor> operands, RemainingResults results, in ExecuteImplAndMaybeLogQueryOfDeath()
[all …]
|
| D | tf_cpurt.h | 88 using ResultType = tensorflow::tfrt_stub::FallbackTensor;
|
| D | tf_jitrt.h | 107 using ResultType = tfrt_stub::FallbackTensor;
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_compat_request_state.h | 46 tfrt::UnRefCountedAsyncValue<tensorflow::tfrt_stub::FallbackTensor>*
52 const tensorflow::tfrt_stub::FallbackTensor& GetResourceAsFallbackTensor( in GetResourceAsFallbackTensor()
66 tfrt::UnRefCountedAsyncValue<tensorflow::tfrt_stub::FallbackTensor>>>
|
| D | kernel_fallback_execute_compat.cc | 523 auto& fallback_tensor = arg->get<tensorflow::tfrt_stub::FallbackTensor>(); in KernelFallbackExecuteOpInternal()
543 tensorflow::tfrt_stub::FallbackTensor>( in KernelFallbackExecuteOpInternal()
547 tensorflow::tfrt_stub::FallbackTensor>( in KernelFallbackExecuteOpInternal()
648 tfrt::Argument<tensorflow::tfrt_stub::FallbackTensor> arg, in FallbackSetResource()
901 tfrt::Argument<tensorflow::tfrt_stub::FallbackTensor> arg, in FallbackCopyTensorIfSmall()
914 tensorflow::tfrt_stub::FallbackTensor>( in FallbackCopyTensorIfSmall()
923 tensorflow::tfrt_stub::FallbackTensor>(fallback_tensor); in FallbackCopyTensorIfSmall()
928 llvm::Expected<tensorflow::tfrt_stub::FallbackTensor> ConstTensorProto( in ConstTensorProto()
941 return tensorflow::tfrt_stub::FallbackTensor(std::move(tensor)); in ConstTensorProto()
|
| D | kernel_fallback_utils.h | 33 tfrt::RepeatedSyncArguments<tfrt_stub::FallbackTensor>&>;
|
| D | kernel_fallback_compat_request_state.cc | 50 tfrt::UnRefCountedAsyncValue<tensorflow::tfrt_stub::FallbackTensor>>( in SetResource()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/ |
| D | runtime_fallback_executor.cc | 66 using ::tensorflow::tfrt_stub::FallbackTensor;
189 auto av = MakeAvailableAsyncValueRef<FallbackTensor>(input_tensor); in Execute()
206 ret_values.push_back(results[i]->get<tfrt_stub::FallbackTensor>().tensor()); in Execute()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 829 tensorflow::tfrt_stub::FallbackTensor>(); in CoreRTTensorHandleToFallbackTensorInternal()
919 tensorflow::tfrt_stub::FallbackTensor>( in CoreRTTensorHandleToFallbackTensorInternal()
940 static llvm::Expected<tensorflow::tfrt_stub::FallbackTensor> ConstDenseTensor( in ConstDenseTensor()
947 return tensorflow::tfrt_stub::FallbackTensor(tensor); in ConstDenseTensor()
950 static llvm::Expected<tensorflow::tfrt_stub::FallbackTensor> ConstStringTensor( in ConstStringTensor()
974 return tensorflow::tfrt_stub::FallbackTensor(tensor); in ConstStringTensor()
1006 auto& tf_tensor = av->get<tensorflow::tfrt_stub::FallbackTensor>().tensor(); in FallbackTensorToCoreRTTensorHandleInternal()
1036 const tensorflow::tfrt_stub::FallbackTensor& input, in Predicate()
1066 const tensorflow::tfrt_stub::FallbackTensor& arg, const tfrt::Chain& ch) { in PrintFallbackTensor()
|
| D | runtime_fallback_batch_tf_opkernels.cc | 481 tfrt::AsyncValueRef<tfrt_stub::FallbackTensor> TFTensorToFallbackTensor( in TFTensorToFallbackTensor()
483 return tfrt::MakeAvailableAsyncValueRef<tfrt_stub::FallbackTensor>(tf_tensor); in TFTensorToFallbackTensor()
596 result->get<tfrt_stub::FallbackTensor>().tensor(); in ProcessFuncBatchImpl()
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.cc | 206 tfrt::MakeAvailableAsyncValueRef<FallbackTensor>(input).release()); in GraphExecutionRunOnFunction()
254 DCHECK(result->IsType<FallbackTensor>()); in GraphExecutionRunOnFunction()
255 const auto& host_tensor = result->get<FallbackTensor>().tensor(); in GraphExecutionRunOnFunction()
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.cc | 237 tfrt::MakeAvailableAsyncValueRef<FallbackTensor>(capture).release()); in RunInitializers()
|