OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:AccessTensor
(Results
1 - 15
of
15
) sorted by null
/external/tensorflow/tensorflow/core/framework/
resource_op_kernel.h
92
auto h = handle_.
AccessTensor
(context)->template flat<string>();
102
context->set_output_ref(0, &mu_, handle_.
AccessTensor
(context));
op_kernel.cc
170
Tensor* PersistentTensor::
AccessTensor
(OpKernelConstruction* context) {
176
Tensor* PersistentTensor::
AccessTensor
(OpKernelContext* context) {
247
Tensor* allocated = out_persistent->
AccessTensor
(this);
692
*out_tensor = out_persistent->
AccessTensor
(this);
695
Tensor* t = out_persistent->
AccessTensor
(this);
[
all
...]
op_kernel.h
208
// order to retrieve the Tensor the caller must use
AccessTensor
which
216
Tensor*
AccessTensor
(OpKernelConstruction* context);
218
Tensor*
AccessTensor
(OpKernelContext* context);
254
// from the PersistentTensor using the
AccessTensor
method. This
271
// out_persistent->
AccessTensor
. The caller does not own out_tensor
773
// the
AccessTensor
method. This ensures that the system is made
[
all
...]
/external/tensorflow/tensorflow/core/kernels/
lookup_table_op.cc
361
key_buckets_.
AccessTensor
(ctx)->template matrix<K>();
363
value_buckets_.
AccessTensor
(ctx)->template matrix<V>();
365
empty_key_.
AccessTensor
(ctx)->template shaped<K, 2>({1, key_size});
442
empty_key_.
AccessTensor
(ctx)->template shaped<K, 2>(
445
key_buckets_.
AccessTensor
(ctx)->template matrix<K>();
456
Tensor key_buckets_tensor = *key_buckets_.
AccessTensor
(ctx);
457
Tensor value_buckets_tensor = *value_buckets_.
AccessTensor
(ctx);
512
key_buckets_.
AccessTensor
(ctx)->template matrix<K>();
514
value_buckets_.
AccessTensor
(ctx)->template matrix<V>();
516
empty_key_.
AccessTensor
(ctx)->template shaped<K, 2>({1, key_size})
[
all
...]
conditional_accumulator_base_op.h
61
ctx->set_output_ref(0, &mu_, accumulator_handle_.
AccessTensor
(ctx));
105
auto h = accumulator_handle_.
AccessTensor
(ctx)->template flat<string>();
lookup_table_op.h
93
auto h = table_handle_.
AccessTensor
(ctx)->template flat<string>();
97
ctx->set_output_ref(0, &mu_, table_handle_.
AccessTensor
(ctx));
queue_ops.cc
462
handle_.
AccessTensor
(context)->flat<string>()(0) = ref.container();
463
handle_.
AccessTensor
(context)->flat<string>()(1) = ref.name();
464
context->set_output_ref(0, &mu_, handle_.
AccessTensor
(context));
priority_queue.cc
67
(*tuple).push_back(*persistent_tensor.
AccessTensor
(ctx));
164
Tensor* priority_tensor = priority_element.
AccessTensor
(ctx);
tensor_array.h
461
Tensor* value_t = value->
AccessTensor
(ctx);
514
Tensor* existing_t = t.tensor.
AccessTensor
(ctx);
barrier_ops.cc
394
ready_tuple.push_back(*element[0].
AccessTensor
(ctx)); // index
398
ready_tuple.push_back(*element[j].
AccessTensor
(ctx));
sparse_tensors_map_ops.cc
93
const Tensor* ix = sp_iter->second.indices.
AccessTensor
(ctx);
94
const Tensor* values = sp_iter->second.values.
AccessTensor
(ctx);
fifo_queue.cc
45
(*tuple).push_back(*queues_[i][0].
AccessTensor
(ctx));
tensor_array_ops.cc
498
ctx->set_output(0, *value.
AccessTensor
(ctx));
630
const Tensor* value_0_t = values[0].
AccessTensor
(ctx);
660
const Tensor* value_t = values[i].
AccessTensor
(ctx);
[
all
...]
random_shuffle_queue_op.cc
121
(*tuple).push_back(*queues_[i][index].
AccessTensor
(ctx));
/external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/
cudnn_rnn_ops.cc
330
return AsDeviceMemory<uint8>(handle_.
AccessTensor
(context_));
[
all
...]
Completed in 375 milliseconds