LostTech.TensorFlow : API Documentation

Type tpu_estimator

Namespace tensorflow_estimator.python.estimator.tpu.tpu_estimator

Methods

Properties

Public static methods

object call_computation(object computation_inputs, object computation, object batch_config)

object call_computation_dyn(object computation_inputs, object computation, object batch_config)

object export_estimator_savedmodel(object estimator, object export_dir_base, PythonFunctionContainer serving_input_receiver_fn, object assets_extra, bool as_text, object checkpoint_path)

object export_estimator_savedmodel_dyn(object estimator, object export_dir_base, object serving_input_receiver_fn, object assets_extra, ImplicitContainer<T> as_text, object checkpoint_path)

ValueTuple<object, _CapturedObject, object> generate_broadcast_enqueue_ops_fn(object ctx, PythonFunctionContainer input_fn, _InputPipeline.InputsStructureRecorder inputs_structure_recorder, object num_hosts)

object generate_broadcast_enqueue_ops_fn_dyn(object ctx, object input_fn, object inputs_structure_recorder, object num_hosts)

ValueTuple<object, _CapturedObject> generate_per_core_enqueue_ops_fn_for_host(object ctx, PythonFunctionContainer input_fn, _InputPipeline.InputsStructureRecorder inputs_structure_recorder, object host_device, int host_id)

object generate_per_core_enqueue_ops_fn_for_host_dyn(object ctx, object input_fn, object inputs_structure_recorder, object host_device, object host_id)

ValueTuple<object, _CapturedObject, object> generate_per_host_enqueue_ops_fn_for_host(object ctx, PythonFunctionContainer input_fn, _InputPipeline.InputsStructureRecorder inputs_structure_recorder, object batch_axis, object device, int host_id)

object generate_per_host_enqueue_ops_fn_for_host_dyn(object ctx, object input_fn, object inputs_structure_recorder, object batch_axis, object device, object host_id)

ValueTuple<object, _CapturedObject, object> generate_per_host_v2_enqueue_ops_fn_for_host(object ctx, PythonFunctionContainer input_fn, _InputPipeline.InputsStructureRecorder inputs_structure_recorder, object device, int host_id)

object generate_per_host_v2_enqueue_ops_fn_for_host_dyn(object ctx, object input_fn, object inputs_structure_recorder, object device, object host_id)

object inference_on_tpu(object computation, object inputs_to_tpu, object num_batch_threads, object max_batch_size, object batch_timeout_micros, object allowed_batch_sizes, int max_enqueued_batches)

object inference_on_tpu_dyn(object computation, object inputs_to_tpu, object num_batch_threads, object max_batch_size, object batch_timeout_micros, object allowed_batch_sizes, ImplicitContainer<T> max_enqueued_batches)

object model_fn_inference_on_tpu(PythonFunctionContainer model_fn, IEnumerable<_DatasetInitializerHook> features, object labels, object config, object params, object batch_config)

object model_fn_inference_on_tpu(PythonFunctionContainer model_fn, object features, IEnumerable<_DatasetInitializerHook> labels, object config, object params, object batch_config)

object model_fn_inference_on_tpu(PythonFunctionContainer model_fn, object features, object labels, object config, object params, object batch_config)

object model_fn_inference_on_tpu(PythonFunctionContainer model_fn, IEnumerable<_DatasetInitializerHook> features, IEnumerable<_DatasetInitializerHook> labels, object config, object params, object batch_config)

object model_fn_inference_on_tpu_dyn(object model_fn, object features, object labels, object config, object params, object batch_config)

Public properties

PythonFunctionContainer _CapturedObject_fn get;

PythonFunctionContainer _CapturingContext_fn get;

PythonFunctionContainer _InputPipeline_fn get;

PythonFunctionContainer _Inputs_fn get;

PythonFunctionContainer _InputsWithStoppingSignals_fn get;

PythonFunctionContainer _ModelFnWrapper_fn get;

PythonFunctionContainer _OpQueueContext_fn get;

PythonFunctionContainer _OpSignalOnceQueueContext_fn get;

PythonFunctionContainer _OutfeedHostCall_fn get;

PythonFunctionContainer _OutfeedHostCallHook_fn get;

PythonFunctionContainer _PaddingSignals_fn get;

PythonFunctionContainer _SetEvalIterationsHook_fn get;

PythonFunctionContainer _SIGNAL_fn get;

PythonFunctionContainer _StoppingPredictHook_fn get;

PythonFunctionContainer _StopSignals_fn get;

PythonFunctionContainer _TPUStopAtStepHook_fn get;

PythonFunctionContainer BatchConfig_fn get;

PythonFunctionContainer call_computation_fn get;

PythonFunctionContainer CatchInvalidHostcallFunctions_fn get;

PythonFunctionContainer ExamplesPerSecondHook_fn get;

PythonFunctionContainer export_estimator_savedmodel_fn get;

PythonFunctionContainer ExportSavedModelApiVersion_fn get;

PythonFunctionContainer generate_broadcast_enqueue_ops_fn_fn get;

PythonFunctionContainer generate_per_core_enqueue_ops_fn_for_host_fn get;

PythonFunctionContainer generate_per_host_enqueue_ops_fn_for_host_fn get;

PythonFunctionContainer generate_per_host_v2_enqueue_ops_fn_for_host_fn get;

PythonFunctionContainer inference_on_tpu_fn get;

PythonFunctionContainer InstallSignalHandlerHook_fn get;

PythonFunctionContainer model_fn_inference_on_tpu_fn get;

PythonFunctionContainer PeriodicLogger_fn get;

PythonFunctionContainer TensorPacker_fn get;

PythonFunctionContainer TPUEstimator_fn get;

PythonFunctionContainer TPUEstimatorSpec_fn get;

PythonFunctionContainer TPUInfeedOutfeedSessionHook_fn get;

PythonFunctionContainer TPUInfeedOutfeedSessionHookForPrediction_fn get;