| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 364 empty_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in MutableDenseHashTable()
366 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in MutableDenseHashTable()
401 key_buckets_.AccessTensor(ctx)->template matrix<K>(); in Find()
403 value_buckets_.AccessTensor(ctx)->template matrix<V>(); in Find()
405 empty_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in Find()
407 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in Find()
502 empty_key_.AccessTensor(ctx)->template shaped<K, 2>( in ImportValues()
505 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>( in ImportValues()
508 key_buckets_.AccessTensor(ctx)->template matrix<K>(); in ImportValues()
520 Tensor key_buckets_tensor = *key_buckets_.AccessTensor(ctx); in ExportValues()
[all …]
|
| D | queue_ops.cc | 69 handle_.AccessTensor(context)->flat<string>()(0) = ref.container(); in Compute()
70 handle_.AccessTensor(context)->flat<string>()(1) = ref.name(); in Compute()
71 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
| D | conditional_accumulator_base_op.h | 63 ctx->set_output_ref(0, &mu_, accumulator_handle_.AccessTensor(ctx)); in Compute()
108 auto h = accumulator_handle_.AccessTensor(ctx)->template flat<string>(); in SetAccumulatorHandle()
|
| D | lookup_table_op.h | 93 auto h = table_handle_.AccessTensor(ctx)->template flat<string>(); in Compute()
97 ctx->set_output_ref(0, &mu_, table_handle_.AccessTensor(ctx)); in Compute()
|
| D | priority_queue.cc | 67 (*tuple).push_back(*persistent_tensor.AccessTensor(ctx)); in DequeueLocked()
164 Tensor* priority_tensor = priority_element.AccessTensor(ctx); in TryEnqueueMany()
|
| D | tensor_array.h | 464 Tensor* value_t = value->AccessTensor(ctx); in LockedWriteOrAggregate()
517 Tensor* existing_t = t.tensor.AccessTensor(ctx); in LockedWriteOrAggregate()
|
| D | sparse_tensors_map_ops.cc | 93 const Tensor* ix = sp_iter->second.indices.AccessTensor(ctx); in RetrieveAndClearSparseTensors()
94 const Tensor* values = sp_iter->second.values.AccessTensor(ctx); in RetrieveAndClearSparseTensors()
|
| D | barrier_ops.cc | 393 ready_tuple.push_back(*element[0].AccessTensor(ctx)); // index in InsertOneLocked()
397 ready_tuple.push_back(*element[j].AccessTensor(ctx)); in InsertOneLocked()
|
| D | tensor_array_ops.cc | 530 ctx->set_output(0, *value.AccessTensor(ctx)); in Compute()
663 const Tensor* value_0_t = values[0].AccessTensor(ctx); in Compute()
693 const Tensor* value_t = values[i].AccessTensor(ctx); in Compute()
884 value_tensors[i] = values[i].AccessTensor(ctx); in Compute()
|
| D | mkl_conv_ops.cc | 928 *cached_filter_data_ptensor_.AccessTensor(context); in Compute()
1397 *cached_filter_data_ptensor_.AccessTensor(context); in IsFilterCacheEmpty()
1410 *cached_filter_data_ptensor_.AccessTensor(context); in CacheFilter()
1435 *cached_filter_data_ptensor_.AccessTensor(context); in GetCachedFilter()
1437 *cached_filter_md_ptensor_.AccessTensor(context); in GetCachedFilter()
|
| D | fifo_queue.cc | 45 (*tuple).push_back(*queues_[i][0].AccessTensor(ctx)); in DequeueLocked()
|
| D | random_shuffle_queue_op.cc | 121 (*tuple).push_back(*queues_[i][index].AccessTensor(ctx)); in DequeueLocked()
|
| D | functional_ops.cc | 656 context->set_output(0, *value_handle_.AccessTensor(context)); in Compute()
|
| D | cudnn_rnn_ops.cc | 469 return AsDeviceMemory<uint8>(handle_.AccessTensor(context_)); in AllocateBytes()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | resource_op_kernel.h | 99 auto h = handle_.AccessTensor(context)->template flat<string>(); in Compute()
110 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
| D | op_kernel.cc | 188 Tensor* PersistentTensor::AccessTensor(OpKernelConstruction* context) { in AccessTensor() function in tensorflow::PersistentTensor
194 Tensor* PersistentTensor::AccessTensor(OpKernelContext* context) { in AccessTensor() function in tensorflow::PersistentTensor
265 Tensor* allocated = out_persistent->AccessTensor(this); in allocate_persistent()
768 *out_tensor = out_persistent->AccessTensor(this); in allocate_persistent()
771 Tensor* t = out_persistent->AccessTensor(this); in allocate_persistent()
|
| D | op_kernel.h | 248 Tensor* AccessTensor(OpKernelConstruction* context);
250 Tensor* AccessTensor(OpKernelContext* context);
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 317 calib_res->device_tensors_.at(i).AccessTensor(ctx); in ExecuteCalibration()
|