Home
last modified time | relevance | path

Searched refs:HandleFromInput (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dsummary_kernels.cc49 ctx, HandleFromInput(ctx, 0), &s, in Compute()
81 ctx, HandleFromInput(ctx, 0), &s, in Compute()
105 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
119 ctx, HandleFromInput(ctx, 0))); in Compute()
131 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
156 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
177 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
200 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
229 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
265 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &s)); in Compute()
[all …]
Dresource_variable_ops.cc132 const ResourceHandle& handle = HandleFromInput(ctx, 0); in Compute()
173 handles[i] = &HandleFromInput(ctx, i); in Compute()
261 LookupResource(ctx, HandleFromInput(ctx, 0), &variable)); in Compute()
305 const ResourceHandle& p = HandleFromInput(ctx, 0); in Compute()
348 context, HandleFromInput(context, 0), &variable, in Compute()
400 context, HandleFromInput(context, 0), &variable, in Compute()
495 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
563 Status s = LookupResource(context, HandleFromInput(context, 0), &variable); in Compute()
618 OP_REQUIRES_OK(c, LookupResource(c, HandleFromInput(c, 0), &v)); in Compute()
760 OP_REQUIRES_OK(c, LookupResource(c, HandleFromInput(c, 0), &v)); in Compute()
Dmutex_ops.cc177 LookupOrCreateResource<Mutex>(c, HandleFromInput(c, 0), &mutex, in ComputeAsync()
180 c, HandleFromInput(c, 0).name()); in ComputeAsync()
Dtraining_op_helpers.h114 if (LookupResource(ctx, HandleFromInput(ctx, input), maybe_resource).ok()) { in GetTrainingVariableMutex()
245 TF_RETURN_IF_ERROR(LookupResource(ctx, HandleFromInput(ctx, input), &var)); in GetInputTensorFromVariable()
Dcount_up_to_op.cc71 LookupResource<Var>(context, HandleFromInput(context, 0), &variable)); in Compute()
Dlookup_util.cc284 TF_RETURN_IF_ERROR(HandleFromInput(ctx, input_name, &handle)); in GetLookupTable()
301 TF_RETURN_IF_ERROR(HandleFromInput(ctx, input_name, &handle)); in GetInitializableLookupTable()
/external/tensorflow/tensorflow/core/kernels/tensor_forest/
Dresource_ops.cc44 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
59 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
78 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
106 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dprediction_ops.cc33 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
/external/tensorflow/tensorflow/contrib/boosted_trees/kernels/
Dmodel_ops.cc58 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
73 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
92 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
116 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
151 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dstats_accumulator_ops.cc372 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
406 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
553 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
595 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
635 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
664 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
693 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
723 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dquantile_ops.cc254 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
472 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
531 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
600 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
646 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dprediction_ops.cc168 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
377 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dstats_ops.cc69 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
91 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
120 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
255 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 1), in Compute()
258 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
366 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 1), in Compute()
369 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
467 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
470 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 1), in Compute()
Dmodel_ops.cc63 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
80 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
104 auto handle = HandleFromInput(context, 0); in Compute()
135 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
186 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
287 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
340 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
388 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
/external/tensorflow/tensorflow/core/kernels/boosted_trees/
Dresource_ops.cc63 CreateResource(context, HandleFromInput(context, 0), result.release()); in Compute()
82 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
145 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
173 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dquantile_ops.cc121 auto status = CreateResource(context, HandleFromInput(context, 0), result); in Compute()
226 HandleFromInput(context, kResourceHandleName, &handle)); in Compute()
286 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
338 HandleFromInput(context, kResourceHandleName, &handle)); in Compute()
393 HandleFromInput(context, kResourceHandleName, &handle)); in Compute()
Dprediction_ops.cc59 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
198 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
302 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
Dtraining_ops.cc58 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
252 OP_REQUIRES_OK(context, LookupResource(context, HandleFromInput(context, 0), in Compute()
/external/tensorflow/tensorflow/contrib/framework/kernels/
Dzero_initializer_op.cc100 ctx, HandleFromInput(ctx, 0), &variable, in Compute()
130 output->scalar<ResourceHandle>()() = HandleFromInput(ctx, 0); in Compute()
/external/tensorflow/tensorflow/core/kernels/data/
Diterator_ops.cc557 ctx, LookupResource(ctx, HandleFromInput(ctx, 1), &iterator_resource)); in Compute()
950 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator), done); in ComputeAsync()
982 OP_REQUIRES_OK(ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator)); in Compute()
1001 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator), done); in ComputeAsync()
1059 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator_resource)); in Compute()
1136 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator_resource)); in Compute()
1155 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &iterator_resource)); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.cc54 context, HandleFromInput(context, 0), &variable, in Compute()
Dxla_launch_util.cc86 [&](int variable_idx) { return &HandleFromInput(ctx, variable_idx); }); in GetVariableInfosFromCtxInputs()
158 tensor.name = HandleFromInput(ctx, variable_indices[i]).name(); in SnapshotResourceVariables()
400 ctx, HandleFromInput(ctx, actual_input_index), &variable, in PopulateOutputs()
/external/tensorflow/tensorflow/core/framework/
Dresource_mgr.cc291 const ResourceHandle& HandleFromInput(OpKernelContext* ctx, int input) { in HandleFromInput() function
295 Status HandleFromInput(OpKernelContext* ctx, StringPiece input, in HandleFromInput() function
/external/tensorflow/tensorflow/contrib/bigtable/kernels/
Dbigtable_sample_keys_dataset_op.cc30 LookupResource(ctx, HandleFromInput(ctx, 0), &resource)); in MakeDataset()
Dbigtable_kernels.cc145 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &client_resource)); in Compute()
242 ctx, LookupResource(ctx, HandleFromInput(ctx, 0), &resource), done); in ComputeAsync()

12