| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_table_op.cc | 386 empty_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in MutableDenseHashTable()
388 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in MutableDenseHashTable()
423 key_buckets_.AccessTensor(ctx)->template matrix<K>(); in Find()
425 value_buckets_.AccessTensor(ctx)->template matrix<V>(); in Find()
427 empty_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in Find()
429 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>({1, key_size}); in Find()
524 empty_key_.AccessTensor(ctx)->template shaped<K, 2>( in ImportValues()
527 deleted_key_.AccessTensor(ctx)->template shaped<K, 2>( in ImportValues()
530 key_buckets_.AccessTensor(ctx)->template matrix<K>(); in ImportValues()
542 Tensor key_buckets_tensor = *key_buckets_.AccessTensor(ctx); in ExportValues()
[all …]
|
| D | queue_ops.cc | 87 handle_.AccessTensor(context)->flat<tstring>()(0) = ref.container(); in Compute()
88 handle_.AccessTensor(context)->flat<tstring>()(1) = ref.name(); in Compute()
89 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
| D | lookup_table_op.h | 97 table_handle_.AccessTensor(ctx)->template scalar<ResourceHandle>(); in Compute()
101 ctx->set_output(0, *table_handle_.AccessTensor(ctx)); in Compute()
104 auto h = table_handle_.AccessTensor(ctx)->template flat<tstring>(); in Compute()
108 ctx->set_output_ref(0, &mu_, table_handle_.AccessTensor(ctx)); in Compute()
|
| D | conditional_accumulator_op.cc | 51 ctx->set_output_ref(0, &mu_, accumulator_handle_.AccessTensor(ctx)); in SetHandleToOutput()
88 auto h = accumulator_handle_.AccessTensor(ctx)->template flat<tstring>(); in SetHandleToOutput()
|
| D | sparse_conditional_accumulator_op.cc | 53 ctx->set_output_ref(0, &mu_, accumulator_handle_.AccessTensor(ctx)); in SetHandleToOutput()
|
| D | priority_queue.cc | 67 (*tuple).push_back(*persistent_tensor.AccessTensor(ctx)); in DequeueLocked()
165 Tensor* priority_tensor = priority_element.AccessTensor(ctx); in TryEnqueueMany()
|
| D | tensor_array.h | 463 Tensor* value_t = value->AccessTensor(ctx); in LockedWriteOrAggregate()
516 Tensor* existing_t = t.tensor.AccessTensor(ctx); in LockedWriteOrAggregate()
|
| D | conditional_accumulator_base_op.h | 116 auto h = accumulator_handle_.AccessTensor(ctx)->template flat<tstring>(); in SetAccumulatorHandle()
|
| 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 | 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 | tensor_array_ops.cc | 526 ctx->set_output(0, *value.AccessTensor(ctx)); in Compute()
658 const Tensor* value_0_t = values[0].AccessTensor(ctx); in Compute()
688 const Tensor* value_t = values[i].AccessTensor(ctx); in Compute()
878 value_tensors[i] = values[i].AccessTensor(ctx); in Compute()
|
| D | fifo_queue.cc | 45 (*tuple).push_back(*queues_[i][0].AccessTensor(ctx)); in DequeueLocked()
|
| D | random_shuffle_queue_op.cc | 122 (*tuple).push_back(*queues_[i][index].AccessTensor(ctx)); in DequeueLocked()
|
| D | functional_ops.cc | 887 context->set_output(0, *value_handle_.AccessTensor(context)); in Compute()
|
| D | cudnn_rnn_ops.cc | 466 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<tstring>(); in Compute()
110 context->set_output_ref(0, &mu_, handle_.AccessTensor(context)); in Compute()
|
| D | op_kernel.cc | 216 Tensor* PersistentTensor::AccessTensor(OpKernelConstruction* context) { in AccessTensor() function in tensorflow::PersistentTensor
222 Tensor* PersistentTensor::AccessTensor(OpKernelContext* context) { in AccessTensor() function in tensorflow::PersistentTensor
313 Tensor* allocated = out_persistent->AccessTensor(this); in allocate_persistent()
831 Tensor* t = out_persistent->AccessTensor(this); in allocate_persistent()
|
| D | op_kernel.h | 241 Tensor* AccessTensor(OpKernelConstruction* context);
243 Tensor* AccessTensor(OpKernelContext* context);
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_ops_common.h | 431 const Tensor& weight_t = *weight_oi_.AccessTensor(context); in CacheWeight()
475 const Tensor& weight_t = *weight_oi_.AccessTensor(context); in GetCachedWeight()
476 const Tensor& weight_md_t = *weight_oi_md_.AccessTensor(context); in GetCachedWeight()
|
| D | mkl_conv_ops.cc | 1084 *cached_filter_data_ptensor_.AccessTensor(context); in IsFilterCacheEmpty()
1097 *cached_filter_data_ptensor_.AccessTensor(context); in CacheFilter()
1140 *cached_filter_data_ptensor_.AccessTensor(context); in GetCachedFilter()
1142 *cached_filter_md_ptensor_.AccessTensor(context); in GetCachedFilter()
1645 *cached_bias_data_ptensor_.AccessTensor(context); in GetCachedBias()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 507 calib_ctx->device_tensors_.at(i).AccessTensor(ctx); in ExecuteCalibration()
|