| /external/tensorflow/tensorflow/core/kernels/ |
| variable_ops.h | 34 explicit Var(DataType dtype) : tensor_(dtype) {}
41 Tensor* tensor() { return &tensor_; }
44 return strings::StrCat(DataTypeString(tensor_.dtype()), "/",
45 tensor_.shape().DebugString());
50 Tensor tensor_; member in class:tensorflow::Var
|
| constant_op.h | 35 Tensor tensor_; member in class:tensorflow::ConstantOp
49 Tensor tensor_; member in class:tensorflow::HostConstantOp
|
| constant_op.cc | 71 tensor_(ctx->output_type(0)) {
75 *proto, AllocatorAttributes(), &tensor_));
77 ctx, ctx->output_type(0) == tensor_.dtype(),
79 DataTypeString(tensor_.dtype()), ") and dtype (",
84 ctx->set_output(0, tensor_);
86 ctx->record_persistent_memory_allocation(tensor_.AllocatedBytes());
133 : OpKernel(ctx), tensor_(ctx->output_type(0)) {
139 ctx, ctx->device()->MakeTensorFromProto(*proto, alloc_attr, &tensor_));
141 ctx, ctx->output_type(0) == tensor_.dtype(),
143 DataTypeString(tensor_.dtype()), ") and dtype ("
[all...] |
| variable_ops.cc | 30 explicit LegacyVar(DataType dtype) : tensor_(dtype) {}
36 Tensor* tensor() { return &tensor_; }
39 return strings::StrCat(DataTypeString(tensor_.dtype()), "/",
40 tensor_.shape().DebugString());
45 Tensor tensor_; member in class:tensorflow::LegacyVar
|
| sparse_cross_op.cc | 105 explicit DenseTensorColumn(const Tensor& tensor) : tensor_(tensor) {}
107 int64 FeatureCount(int64 batch) const override { return tensor_.dim_size(1); }
114 const Tensor& tensor_; member in class:tensorflow::__anon39924::DenseTensorColumn
120 if (DT_STRING == tensor_.dtype())
121 return Fingerprint64(tensor_.matrix<string>()(batch, n));
122 return tensor_.matrix<int64>()(batch, n);
128 if (DT_STRING == tensor_.dtype()) return tensor_.matrix<string>()(batch, n);
129 return std::to_string(tensor_.matrix<int64>()(batch, n));
135 return tensor_.matrix<string>()(batch, n)
[all...] |
| /external/tensorflow/tensorflow/core/profiler/internal/ |
| tfprof_tensor.cc | 36 tfprof_tensor_pb_.set_dtype(tensor_->dtype());
38 switch (tensor_->dtype()) {
43 if (tensor_->dtype() == DataType::DT_FLOAT) {
45 } else if (tensor_->dtype() == DataType::DT_DOUBLE) {
55 if (tensor_->dtype() == DataType::DT_INT32) {
57 } else if (tensor_->dtype() == DataType::DT_INT64) {
71 fprintf(stderr, "Not Supported type %d\n", tensor_->dtype());
|
| tfprof_tensor.h | 39 : tensor_(std::move(tensor)) {
89 if (tensor_->dims() == 0 && values.size() == 1) {
115 for (int i = 0; i < tensor_->dim_size(depth); i++) {
117 if (depth == tensor_->dims() - 1) {
160 if (tensor_->NumElements() > kTFProfTensorMaxWarnLen) {
163 auto values = tensor_->flat<T>();
164 for (int64 i = 0; i < tensor_->NumElements(); i++) {
170 std::unique_ptr<Tensor> tensor_; member in class:tensorflow::tfprof::TFProfTensor
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| tensor_coding.h | 42 // Clear just tensor_ and meta_ members without setting allocation
83 const Tensor& tensor() const { return tensor_; }
101 Tensor tensor_; member in class:tensorflow::TensorResponse
|
| tensor_coding.cc | 39 tensor_ = Tensor();
57 if (!tensor_.FromProto(allocator_, meta_.tensor())) {
61 s = device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_);
78 tensor_ = std::move(t);
91 device_->MakeTensorFromProto(meta_.tensor(), alloc_attrs_, &tensor_);
159 tensor_ = std::move(t);
206 tensor_ = std::move(t);
282 tensor_ = std::move(parsed);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| ops.h | 214 tensor_(init.tensor) {}
218 tensor_(t) {}
228 tensor_ = Initializer(init).tensor;
241 const Tensor& tensor() const { return tensor_; }
246 Tensor tensor_; member in class:tensorflow::Input
|
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| sparse_feature_cross_kernel.cc | 105 explicit DenseTensorColumn(const Tensor& tensor) : tensor_(tensor) {}
107 int64 FeatureCount(int64 batch) const override { return tensor_.dim_size(1); }
114 const Tensor& tensor_; member in class:tensorflow::__anon39179::DenseTensorColumn
120 if (DT_STRING == tensor_.dtype())
121 return Fingerprint64(tensor_.matrix<string>()(batch, n));
122 return tensor_.matrix<int64>()(batch, n);
128 if (DT_STRING == tensor_.dtype()) return tensor_.matrix<string>()(batch, n);
129 return std::to_string(tensor_.matrix<int64>()(batch, n));
135 return tensor_.matrix<string>()(batch, n)
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| op_kernel.h | 213 explicit PersistentTensor(const Tensor& tensor) : tensor_(tensor) {}
222 bool IsInitialized() const { return tensor_.IsInitialized(); }
224 int64 NumElements() const { return tensor_.NumElements(); }
226 int64 AllocatedBytes() const { return tensor_.AllocatedBytes(); }
229 Tensor tensor_; member in class:tensorflow::PersistentTensor
[all...] |
| op_kernel.cc | 173 return &tensor_;
177 context->NotifyUseOfPersistentTensor(tensor_);
178 return &tensor_;
[all...] |
| /external/tensorflow/tensorflow/contrib/tensorrt/convert/ |
| convert_nodes.cc | 170 : tensor_(tensor), weights_(DT_FLOAT), variant_(TRT_NODE_TENSOR) {}
172 : tensor_(nullptr), weights_(weights), variant_(TRT_NODE_WEIGHTS) {}
174 : tensor_(rhs.tensor_), weights_(rhs.weights_), variant_(rhs.variant_) {}
182 return tensor_;
186 return tensor_;
205 nvinfer1::ITensor* tensor_; member in class:tensorflow::tensorrt::convert::__anon39442::TRT_TensorOrWeights
[all...] |
| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma.cc | [all...] |
| rdma.h | 363 Tensor* tensor_ = nullptr; member in class:tensorflow::RdmaTensorResponse
|
