Home
last modified time | relevance | path

Searched refs:c_status (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc305 OwnedTFStatus c_status(TF_NewStatus()); in SynchronizeAllActivity() local
306 stream_executor_->synchronize_all_activity(&device_, c_status.get()); in SynchronizeAllActivity()
307 if (TF_GetCode(c_status.get()) != TF_OK) { in SynchronizeAllActivity()
308 LOG(ERROR) << TF_Message(c_status.get()); in SynchronizeAllActivity()
327 OwnedTFStatus c_status(TF_NewStatus()); in SynchronousMemcpy() local
330 size, c_status.get()); in SynchronousMemcpy()
331 return StatusFromTF_Status(c_status.get()); in SynchronousMemcpy()
336 OwnedTFStatus c_status(TF_NewStatus()); in SynchronousMemcpy() local
339 size, c_status.get()); in SynchronousMemcpy()
340 return StatusFromTF_Status(c_status.get()); in SynchronousMemcpy()
[all …]
Dstream_executor_internal.h63 tensorflow::TF_StatusPtr c_status(TF_NewStatus()); in VisibleDeviceCount()
65 c_status.get()); in VisibleDeviceCount()
66 if (TF_GetCode(c_status.get()) != TF_OK) { in VisibleDeviceCount()
67 LOG(ERROR) << TF_Message(c_status.get()); in VisibleDeviceCount()
114 tensorflow::TF_StatusPtr c_status(TF_NewStatus()); in Create()
115 stream_executor_->create_stream(device_, &stream_handle_, c_status.get()); in Create()
116 port::Status s = tensorflow::StatusFromTF_Status(c_status.get()); in Create()
144 tensorflow::TF_StatusPtr c_status(TF_NewStatus()); in Create()
145 stream_executor_->create_event(device_, &event_handle_, c_status.get()); in Create()
146 return tensorflow::StatusFromTF_Status(c_status.get()); in Create()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dstatus_helper.h26 : c_status(tensorflow::tpu::ExecutorApiFn()->TpuStatus_NewFn()) {} in StatusHelper()
29 tensorflow::tpu::ExecutorApiFn()->TpuStatus_FreeFn(c_status); in ~StatusHelper()
33 TF_Status* const c_status) { in FromC() argument
34 if (tensorflow::tpu::ExecutorApiFn()->TpuStatus_OkFn(c_status)) { in FromC()
39 tensorflow::tpu::ExecutorApiFn()->TpuStatus_CodeFn(c_status)), in FromC()
40 tensorflow::tpu::ExecutorApiFn()->TpuStatus_MessageFn(c_status)); in FromC()
45 return tensorflow::tpu::ExecutorApiFn()->TpuStatus_OkFn(c_status); in ok()
49 return FromC(c_status); in status()
52 TF_Status* const c_status; // NOLINT variable
Dtpu_executor.cc46 status.c_status); in Init()
66 executor_, get_stream(stream->implementation()), status.c_status); in BlockHostUntilDone()
73 status.c_status); in BlockUntilDoneOrFailed()
80 executor_, get_stream(stream->implementation()), status.c_status); in GetStatus()
146 status.c_status); in RecordEvent()
156 status.c_status); in WaitForEvent()
250 executor_, infeed_queue_index, status.c_status); in WaitForInfeedReady()
257 executor_, outfeed_queue_index, status.c_status); in WaitForOutfeedReady()
266 status.c_status); in DequeueOutfeed()
275 status.c_status); in EnqueueInfeed()
[all …]
Dtpu_node_context.cc31 tpu::OpsApiFn()->TpuNodeContext_CreateFn(device_ordinal, status.c_status); in Create()
48 tpu::OpsApiFn()->TpuNodeContext_StopChipHeartbeatsFn(status.c_status); in StopChipHeartbeats()
55 tpu::OpsApiFn()->TpuNodeContext_CloseTpuHostFn(status.c_status); in CloseTpuHost()
62 tpu::OpsApiFn()->TpuNodeContext_InitializeFn(device_ordinal, status.c_status); in Initialize()
Dtpu_transfer_manager.cc85 &c_literal, &c_device_buffer, status.c_status); in TransferLiteralToDeviceAsync()
100 manager_, tpu_executor->se_executor(), &c_literal, status.c_status); in TransferLiteralToInfeed()
127 buffers_size.data(), buffers_size.size(), status.c_status); in TransferBuffersToInfeed()
144 status.c_status); in TransferLiteralFromOutfeed()
163 manager_, se.data(), executor.size(), status.c_status); in ResetDevices()
233 manager_, &c_host_shape, &c_output, status.c_status); in ChooseCompactLayoutForShape()
292 elements_bases, elements.size(), &c_shape, &region_base, status.c_status); in WriteSingleTupleIndexTable()
312 status.c_status); in LinearizeToBuffers()
340 &c_device_buffer, c_device_shape, &c_updated_shape, status.c_status); in ReadDynamicShapes()
Dtpu_stream.h51 size, status.c_status); in EnqueueTransferHostToDevice()
61 status.c_status); in EnqueueTransferDeviceToHost()
72 ApiConverter::ToC(recv_buffer), status.c_status); in EnqueueOnTpuDeviceSendRecvLocal()
Dtpu_executable.cc112 &se_execution_output, status.c_status); in ExecuteAsyncOnStream()
171 status.c_status); in Serialize()
185 status.c_status); in Serialize()
198 &se_executable, status.c_status); in Deserialize()
Dtpu_platform.cc62 platform_, options_size, options_key, options_value, status.c_status); in Initialize()
99 platform_, c_config, status.c_status); in GetUncachedExecutor()
Dtpu_op_executable.cc94 params.status = status.c_status; in LoadProgramAndEnqueueToStream()
/external/tensorflow/tensorflow/core/profiler/backends/tpu/
Dtpu_tracer.cc58 tpu::OpsApiFn()->TpuProfiler_CreateFn(&tpu_profiler_, status.c_status); in TpuTracer()
70 tpu::OpsApiFn()->TpuProfiler_StartFn(tpu_profiler_, status.c_status); in Start()
80 tpu::OpsApiFn()->TpuProfiler_StopFn(tpu_profiler_, status.c_status); in Stop()
92 tpu::OpsApiFn()->TpuProfiler_CollectDataFn(tpu_profiler_, status.c_status, in CollectData()
98 tpu::OpsApiFn()->TpuProfiler_CollectDataFn(tpu_profiler_, status.c_status, in CollectData()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_program_group.cc42 xla_tpu_program, &serialized_executable_info, status.c_status); in ConstructExecutableInfo()
58 xla_tpu_program, &serialized_host_transfer_info, status.c_status); in ConstructHostTransferInfo()
74 xla_tpu_program, &serialized_hlo_metadata, status.c_status); in ConstructHloMetadata()
147 OpsApiFn()->TpuProgram_UnloadAndDestroyFn(tpu_program, status.c_status); in UnloadAndDestroyPrograms()
274 &count, status.c_status); in CompileAndBuild()
308 &count, status.c_status); in CompileAndBuild()
352 rpc_response_protos[i], xla_tpu_program, status.c_status); in DeserializeFromRpcResponseProtos()
370 executable, status.c_status); in SerializeExecutable()
380 tpu_programs_[index], compiler_metadata, status.c_status); in SerializeCompilerMetadata()
Dtpu_embedding_configuration_ops.cc74 params.status = status.c_status; in Compute()
128 params.status = status.c_status; in Compute()
188 params.status = status.c_status; in Compute()
259 params.status = status.c_status; in Compute()
315 params.status = status.c_status; in Compute()
356 params.status = status.c_status; in Compute()
394 params.status = status.c_status; in Compute()
Dtpu_embedding_ops.cc79 recv_activation_params.status = status.c_status; in Compile()
156 recv_deduplication_params.status = status.c_status; in Compile()
251 send_gradients_params.status = status.c_status; in Compile()
Dtpu_embedding_enqueue_ops.cc86 fixed_state_create_params.status = status.c_status; in EnqueueTPUEmbeddingArbitraryTensorBatchOp()
160 params.status = status.c_status; in Compute()
161 params.status = status.c_status; in Compute()
Dtpu_embedding_load_retrieve_ops.cc230 status.c_status); in Compute()
359 status.c_status); in Compute()
/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
Dpjrt_c_api_wrapper_impl.h149 #define _PJRT_ASSIGN_OR_RETURN_IMPL(statusor, lhs, rexpr, c_status) \ argument
152 PJRT_Error* c_status = new PJRT_Error(); \
153 c_status->status = statusor.status(); \
154 return c_status; \
/external/tensorflow/tensorflow/c/experimental/grappler/
Dgrappler.cc88 OwnedTFStatus c_status(TF_NewStatus()); in Optimize() local
95 optimized_graph_buf.get(), c_status.get()); in Optimize()
96 TF_RETURN_IF_ERROR(tensorflow::StatusFromTF_Status(c_status.get())); in Optimize()
167 OwnedTFStatus c_status(TF_NewStatus()); in InitGraphPlugin() local
168 init_fn(&params, c_status.get()); in InitGraphPlugin()
169 TF_RETURN_IF_ERROR(tensorflow::StatusFromTF_Status(c_status.get())); in InitGraphPlugin()
/external/tensorflow/tensorflow/c/experimental/filesystem/
Dmodular_filesystem_registration.cc254 TF_Status* c_status = TF_NewStatus(); in RegisterFileSystem() local
256 core_filesystem_ops->init(filesystem.get(), c_status); in RegisterFileSystem()
257 status = Status(c_status->status); in RegisterFileSystem()
258 TF_DeleteStatus(c_status); in RegisterFileSystem()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtpu_computation_placer.cc49 status.c_status); in AssignDevices()
64 status.c_status); in AssignLocalDevices()
/external/tensorflow/tensorflow/c/
Dops.cc94 TF_Status* c_status = TF_NewStatus(); in TF_OpDefinitionBuilderSetShapeInferenceFunction() local
96 shape_inference_func(c_ctx, c_status); in TF_OpDefinitionBuilderSetShapeInferenceFunction()
97 tensorflow::Status result = ::tensorflow::StatusFromTF_Status(c_status); in TF_OpDefinitionBuilderSetShapeInferenceFunction()
98 TF_DeleteStatus(c_status); in TF_OpDefinitionBuilderSetShapeInferenceFunction()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device_util.h47 #define RETURN_C_STATUS_IF_NOT_OK(cpp_status, c_status) \ argument
51 RETURN_STATUS((c_status), \
59 #define ASSIGN_OR_RETURN_C_STATUS(var, cpp_status, c_status) \ argument
62 var, cpp_status, c_status)
64 #define ASSIGN_OR_RETURN_C_STATUS_IMPL(statusor, var, cpp_status, c_status) \ argument
66 RETURN_C_STATUS_IF_NOT_OK(statusor.status(), (c_status)); \
/external/tensorflow/tensorflow/core/tpu/
Dtpu_on_demand_compiler.cc72 &allocator, &result, status.c_status); in RunHloPasses()
101 &allocator, &result, status.c_status); in RunBackend()
154 &allocator, se_executables, status.c_status); in Compile()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_op_executable.cc94 params.status = status.c_status; in LoadProgramAndEnqueueToStream()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc67 status.c_status); in TpuShapeRepresentation()
97 &se_shape.value, &tpu_shape.value, status.c_status); in TpuPaddedShapeFn()

12