Home
last modified time | relevance | path

Searched refs:resource_handle (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py67 def __init__(self, resource_handle, create_op, num_streams, name): argument
68 self._resource_handle = resource_handle
115 resources.register_resource(self.resource_handle, self._init_op,
118 self.resource_handle, self._init_op, self._num_streams,
119 self.resource_handle.name)
127 return create_quantile_stream_resource(self.resource_handle, self._eps,
137 return is_quantile_resource_initialized(self.resource_handle)
149 summary_op = quantile_add_summaries(self.resource_handle, summaries)
153 return quantile_flush(self.resource_handle, self._num_quantiles)
156 return get_bucket_boundaries(self.resource_handle, self._num_streams)
[all …]
Dlookup_ops.py192 with ops.name_scope(name, "%s_Size" % self.name, [self.resource_handle]):
193 return gen_lookup_ops.lookup_table_size_v2(self.resource_handle)
221 (self.resource_handle, key_tensor, self._default_value)):
223 self.resource_handle, key_tensor, self._default_value)
305 with ops.name_scope(name, "%s_Export" % self.name, [self.resource_handle]):
307 self.resource_handle, self._key_dtype, self._value_dtype)
412 self._name, values=(table.resource_handle, self._keys, self._values)):
415 table.resource_handle, self._keys, self._values)
418 init_op = gen_lookup_ops.initialize_table_v2(table.resource_handle,
582 with ops.name_scope(self._name, "text_file_init", (table.resource_handle,)):
[all …]
Dsession_ops.py36 def encode_resource_handle(resource_handle): argument
38 return np.asarray(bytearray(resource_handle.SerializeToString()),
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dquantile_ops.py45 def __init__(self, resource_handle, create_op, name): argument
46 self._resource_handle = resource_handle
49 gen_quantile_ops.quantile_accumulator_serialize(resource_handle))
126 resources.register_resource(self.resource_handle, self._init_op,
128 self._saveable = QuantileAccumulatorSaveable(self.resource_handle,
138 self.resource_handle,
153 self.resource_handle)
162 quantile_accumulator_handles=[self.resource_handle],
169 resource_handle=self.resource_handle,
194 quantile_accumulator_handles=[self.resource_handle],
[all …]
Dstats_accumulator_ops.py38 def __init__(self, resource_handle, create_op, is_scalar, name): argument
40 self._resource_handle = resource_handle
150 resources.register_resource(self.resource_handle, self.initializer,
153 self.resource_handle, self.initializer, self._is_scalar, name)
169 self.resource_handle, self._stamp_token)
172 self.resource_handle, self._stamp_token,
184 self.resource_handle)
187 self.resource_handle)
202 [self.resource_handle], stamp_token, [partition_ids], [feature_ids],
206 [self.resource_handle], stamp_token, [partition_ids], [feature_ids],
[all …]
Dmodel_ops.py105 self.resource_handle, self._stamp_token, self._tree_ensemble_config)
114 return gen_model_ops.tree_ensemble_is_initialized_op(self.resource_handle)
118 self.resource_handle.op.name + "/tree_ensemble_variable":
121 tree_ensemble_handle=self.resource_handle,
145 resource_handle = tree_ensemble_var.resource_handle
149 saveable = TreeEnsembleVariableSavable(resource_handle, create_op,
150 resource_handle.name)
152 resources.register_resource(resource_handle, create_op, is_initialized_op)
153 return resource_handle
Dbatch_ops_utils.py48 def __init__(self, resource_handle, op, **kwargs): argument
49 self.resource_handle = resource_handle
56 return self.resource_handle.device, self.op
95 resource_handles.append(op.resource_handle)
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/
Dstats_ops.py117 self.resource_handle,
128 return gen_stats_ops.fertile_stats_is_initialized_op(self.resource_handle)
137 stats_handle=self.resource_handle,
157 resource_handle = fertile_stats_var.resource_handle
163 "fertile_stats_variable"](name=resource_handle.name))
165 resources.register_resource(resource_handle, create_op, is_initialized_op)
166 return resource_handle
Dmodel_ops.py122 self.resource_handle,
133 return gen_model_ops.tree_is_initialized_op(self.resource_handle)
142 tree_handle=self.resource_handle,
163 resource_handle = tree_var.resource_handle
168 name=resource_handle.name)
170 resources.register_resource(resource_handle, create_op, is_initialized_op)
171 return resource_handle
/external/tensorflow/tensorflow/python/kernel_tests/boosted_trees/
Dprediction_ops_test.py38 tree_ensemble_handle = tree_ensemble.resource_handle
108 tree_ensemble_handle = tree_ensemble.resource_handle
178 tree_ensemble_handle = tree_ensemble.resource_handle
289 tree_ensemble_handle = tree_ensemble.resource_handle
419 tree_ensemble_handle = tree_ensemble.resource_handle
502 tree_ensemble_handle = tree_ensemble.resource_handle
624 tree_ensemble_handle = tree_ensemble.resource_handle
767 tree_ensemble_handle = tree_ensemble.resource_handle
842 tree_ensemble_handle = tree_ensemble.resource_handle
883 tree_ensemble_handle = tree_ensemble.resource_handle
[all …]
Dtraining_ops_test.py39 tree_ensemble_handle = tree_ensemble.resource_handle
149 tree_ensemble_handle = tree_ensemble.resource_handle
229 tree_ensemble_handle = tree_ensemble.resource_handle
424 tree_ensemble_handle = tree_ensemble.resource_handle
564 tree_ensemble_handle = tree_ensemble.resource_handle
724 tree_ensemble_handle = tree_ensemble.resource_handle
836 tree_ensemble_handle = tree_ensemble.resource_handle
935 tree_ensemble_handle = tree_ensemble.resource_handle
1271 tree_ensemble_handle = tree_ensemble.resource_handle
1454 tree_ensemble_handle = tree_ensemble.resource_handle
/external/tensorflow/tensorflow/python/training/tracking/
Dtracking_test.py231 self.assertEqual("test1", resource_tracker.resources[0].resource_handle)
232 self.assertEqual("test2", resource_tracker.resources[1].resource_handle)
244 self.assertEqual("test1", resource_tracker1.resources[0].resource_handle)
246 self.assertEqual("test2", resource_tracker2.resources[0].resource_handle)
260 self.assertEqual("test1", resource_tracker1.resources[0].resource_handle)
262 self.assertEqual("test2", resource_tracker2.resources[0].resource_handle)
264 self.assertEqual("test1", resource_tracker.resources[0].resource_handle)
265 self.assertEqual("test2", resource_tracker.resources[1].resource_handle)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_IteratorToStringHandle.pbtxt4 name: "resource_handle"
15 summary: "Converts the given `resource_handle` representing an iterator to a string."
Dapi_def_SerializeIterator.pbtxt4 name: "resource_handle"
16 summary: "Converts the given `resource_handle` representing an iterator to a variant tensor."
Dapi_def_DeserializeIterator.pbtxt4 name: "resource_handle"
Dapi_def_IteratorFromStringHandle.pbtxt10 name: "resource_handle"
/external/tensorflow/tensorflow/core/kernels/
Dsession_ops.cc52 ResourceHandle resource_handle = MakeResourceHandle<Tensor>( in Compute() local
55 resource_handle.set_maybe_type_name( in Compute()
57 handle->scalar<ResourceHandle>()() = resource_handle; in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc622 ResourceHandle resource_handle; in Compute() local
625 resource_handle.ParseFromString(string_handle_t.scalar<string>()()), in Compute()
630 ctx, resource_handle.device() == ctx->device()->attributes().name(), in Compute()
634 resource_handle.device(), "\"")); in Compute()
639 OP_REQUIRES_OK(ctx, LookupResource(ctx, resource_handle, &resource)); in Compute()
653 resource_handle_t->scalar<ResourceHandle>()() = resource_handle; in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/lib/learner/batch/
Dordinal_split_handler.py145 with ops.colocate_with(self._stats_accumulator.resource_handle):
271 handler(self._quantile_accumulator.resource_handle,
272 self._stats_accumulator.resource_handle, stamp_token,
451 handler(self._quantile_accumulator.resource_handle,
452 self._stats_accumulator.resource_handle, stamp_token,
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dvariable_ops.cc132 xla::XlaOp resource_handle; in Compile() local
134 &resource_handle)); in Compile()
141 ctx, XlaGather(resource_handle, resource_shape, indices, indices_shape, in Compile()
/external/tensorflow/tensorflow/contrib/lookup/
Dlookup_ops.py360 [self.resource_handle]) as name:
362 self.resource_handle, self._key_dtype, self._value_dtype, name=name)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dset_stats_aggregator_dataset_op.cc98 const Tensor& resource_handle, in Dataset() argument
103 resource_handle_(resource_handle), in Dataset()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.lookup.-static-vocabulary-table.pbtxt17 name: "resource_handle"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.lookup.-static-vocabulary-table.pbtxt22 name: "resource_handle"
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_core_framework.cmake156 "tensorflow/core/framework/resource_handle.proto"
207 "${tensorflow_source_dir}/tensorflow/core/framework/resource_handle.h"
208 "${tensorflow_source_dir}/tensorflow/core/framework/resource_handle.cc")
315 "${tensorflow_source_dir}/tensorflow/core/framework/resource_handle.cc"

123