HomeSort by relevance Sort by last modified time
    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