| /external/tensorflow/tensorflow/python/ops/ |
| D | boosted_trees_ops.py | 73 def __init__(self, resource_handle, create_op, num_streams, name): argument
74 self._resource_handle = resource_handle
121 resources.register_resource(self.resource_handle, self._init_op,
124 self.resource_handle, self._init_op, self._num_streams,
125 self.resource_handle.name)
133 return create_quantile_stream_resource(self.resource_handle, self._eps,
143 return is_quantile_resource_initialized(self.resource_handle)
155 summary_op = quantile_add_summaries(self.resource_handle, summaries)
159 return quantile_flush(self.resource_handle, self._num_quantiles)
162 return get_bucket_boundaries(self.resource_handle, self._num_streams)
[all …]
|
| D | lookup_ops.py | 204 with ops.name_scope(name, "%s_Size" % self.name, [self.resource_handle]):
205 return gen_lookup_ops.lookup_table_size_v2(self.resource_handle)
235 (self.resource_handle, key_tensor, self._default_value)):
236 values = gen_lookup_ops.lookup_table_find_v2(self.resource_handle,
343 with ops.name_scope(name, "%s_Export" % self.name, [self.resource_handle]):
345 self.resource_handle, self._key_dtype, self._value_dtype)
492 table.resource_handle, self.dataset._variant_tensor) # pylint: disable=protected-access
558 self._name, values=(table.resource_handle, self._keys, self._values)):
559 init_op = gen_lookup_ops.lookup_table_import_v2(table.resource_handle,
747 with ops.name_scope(self._name, "text_file_init", (table.resource_handle,)):
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_op_interfaces.h | 83 friend ::llvm::hash_code hash_value(const ResourceHandle& resource_handle);
92 inline ::llvm::hash_code hash_value(const ResourceHandle& resource_handle) { in hash_value() argument
93 return ::llvm::hash_combine(resource_handle.container, resource_handle.name, in hash_value()
94 resource_handle.device, resource_handle.op); in hash_value()
124 const mlir::TF::ResourceHandle& resource_handle) {
125 return mlir::TF::hash_value(resource_handle);
|
| /external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/ |
| D | prediction_ops_test.py | 40 tree_ensemble_handle = tree_ensemble.resource_handle
74 tree_ensemble_handle = tree_ensemble.resource_handle
148 tree_ensemble_handle = tree_ensemble.resource_handle
217 tree_ensemble_handle = tree_ensemble.resource_handle
289 tree_ensemble_handle = tree_ensemble.resource_handle
360 tree_ensemble_handle = tree_ensemble.resource_handle
434 tree_ensemble_handle = tree_ensemble.resource_handle
522 tree_ensemble_handle = tree_ensemble.resource_handle
636 tree_ensemble_handle = tree_ensemble.resource_handle
787 tree_ensemble_handle = tree_ensemble.resource_handle
[all …]
|
| D | training_ops_test.py | 43 tree_ensemble_handle = tree_ensemble.resource_handle
153 tree_ensemble_handle = tree_ensemble.resource_handle
262 tree_ensemble_handle = tree_ensemble.resource_handle
371 tree_ensemble_handle = tree_ensemble.resource_handle
505 tree_ensemble_handle = tree_ensemble.resource_handle
585 tree_ensemble_handle = tree_ensemble.resource_handle
772 tree_ensemble_handle = tree_ensemble.resource_handle
981 tree_ensemble_handle = tree_ensemble.resource_handle
1193 tree_ensemble_handle = tree_ensemble.resource_handle
1442 tree_ensemble_handle = tree_ensemble.resource_handle
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | tracking_test.py | 169 self.assertEqual("test1", resource_tracker.resources[0].resource_handle)
170 self.assertEqual("test2", resource_tracker.resources[1].resource_handle)
182 self.assertEqual("test1", resource_tracker1.resources[0].resource_handle)
184 self.assertEqual("test2", resource_tracker2.resources[0].resource_handle)
198 self.assertEqual("test1", resource_tracker1.resources[0].resource_handle)
200 self.assertEqual("test2", resource_tracker2.resources[0].resource_handle)
202 self.assertEqual("test1", resource_tracker.resources[0].resource_handle)
203 self.assertEqual("test2", resource_tracker.resources[1].resource_handle)
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | partially_revived_objects.cc | 124 if (resource.resource_handle == nullptr) { in TensorHandleFromNode()
129 *handle = resource.resource_handle.get(); in TensorHandleFromNode()
412 AbstractTensorHandle* resource_handle = nullptr; in CreateAllResourceHandles() local
415 absl::MakeSpan(&resource_handle, num_retvals), &num_retvals)); in CreateAllResourceHandles()
416 AbstractTensorHandlePtr owned_resource_handle(resource_handle); in CreateAllResourceHandles()
424 resource.resource_handle = std::move(result); in CreateAllResourceHandles()
474 if (resource_revival_state.resource_handle == nullptr) { in BuildResources()
486 std::move(resource_revival_state.resource_handle))); in BuildResources()
|
| D | restored_resource.cc | 50 ImmediateTensorHandlePtr resource_handle) in RestoredResource() argument
51 : TensorHandleConvertible(std::move(resource_handle)), in RestoredResource()
|
| D | restored_resource_revival_state.h | 33 ImmediateTensorHandlePtr resource_handle = nullptr; member
|
| D | restored_resource.h | 68 ImmediateTensorHandlePtr resource_handle);
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SerializeIterator.pbtxt | 4 name: "resource_handle"
16 name: "resource_handle"
|
| D | IteratorFromStringHandle.pbtxt | 8 name: "resource_handle"
20 name: "resource_handle"
|
| D | DeserializeIterator.pbtxt | 4 name: "resource_handle"
|
| D | IteratorToStringHandle.pbtxt | 4 name: "resource_handle"
|
| D | IteratorFromStringHandleV2.pbtxt | 8 name: "resource_handle"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SerializeIterator.pbtxt | 4 name: "resource_handle"
16 name: "resource_handle"
|
| D | IteratorFromStringHandle.pbtxt | 8 name: "resource_handle"
20 name: "resource_handle"
|
| D | DeserializeIterator.pbtxt | 4 name: "resource_handle"
|
| D | IteratorToStringHandle.pbtxt | 4 name: "resource_handle"
|
| D | IteratorFromStringHandleV2.pbtxt | 8 name: "resource_handle"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_IteratorToStringHandle.pbtxt | 4 name: "resource_handle"
15 summary: "Converts the given `resource_handle` representing an iterator to a string."
|
| D | api_def_SerializeIterator.pbtxt | 4 name: "resource_handle"
16 summary: "Converts the given `resource_handle` representing an iterator to a variant tensor."
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | session_ops.cc | 58 ResourceHandle resource_handle = MakeResourceHandle<Tensor>( in Compute() local
61 resource_handle.set_maybe_type_name( in Compute()
63 handle->scalar<ResourceHandle>()() = resource_handle; in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lift_variables.cc | 102 const ResourceHandle& resource_handle = in LiftVariablesFromSession() local
106 if (!(device_manager->LookupDevice(resource_handle.device(), &device) in LiftVariablesFromSession()
113 ->Lookup(resource_handle.container(), resource_handle.name(), in LiftVariablesFromSession()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shuffle_dataset_op.cc | 445 ResourceHandle&& resource_handle) in Dataset() argument
448 resource_handle_(std::move(resource_handle)), in Dataset()
503 ResourceHandle&& resource_handle, bool owns_resource) in DatasetV2() argument
507 resource_handle_(std::move(resource_handle)), in DatasetV2()
557 ResourceHandle&& resource_handle, bool owns_resource) in DatasetV3() argument
561 resource_handle_(std::move(resource_handle)), in DatasetV3()
751 ResourceHandle&& resource_handle) in Dataset() argument
754 resource_handle_(std::move(resource_handle)), in Dataset()
806 ResourceHandle&& resource_handle, bool owns_resource) in DatasetV2() argument
810 resource_handle_(std::move(resource_handle)), in DatasetV2()
|