| /external/tensorflow/tensorflow/python/framework/ |
| fast_tensor_util.pyx | 11 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray):
15 tensor_proto.float_val.append(nparray[i])
19 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray):
23 tensor_proto.double_val.append(nparray[i])
27 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray):
31 tensor_proto.int_val.append(nparray[i])
34 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray):
38 tensor_proto.uint32_val.append(nparray[i])
41 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray):
45 tensor_proto.int64_val.append(nparray[i]
[all...] |
| tensor_util.py | 49 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values):
50 tensor_proto.half_val.extend(
59 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values):
60 tensor_proto.half_val.extend(
115 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values):
116 tensor_proto.float_val.extend([np.asscalar(x) for x in proto_values])
118 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values):
119 tensor_proto.double_val.extend([np.asscalar(x) for x in proto_values])
121 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values):
122 tensor_proto.int_val.extend([np.asscalar(x) for x in proto_values]
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| unique_op_test.cc | 39 TensorProto tensor_proto; local
40 tensor_proto.set_dtype(DT_INT32);
41 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim);
42 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false);
45 tensor_proto.add_int_val(int_val);
47 return tensor_proto;
52 TensorProto tensor_proto; local
53 tensor_proto.set_dtype(DT_INT32);
54 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim);
55 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false)
106 TensorProto tensor_proto; local
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| threadpool_device.cc | 69 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs,
71 if (tensor_proto.dtype() > 0 && tensor_proto.dtype() <= DataType_MAX) {
72 Tensor parsed(tensor_proto.dtype());
73 if (parsed.FromProto(cpu_allocator(), tensor_proto)) {
79 ProtoDebugString(tensor_proto));
|
| threadpool_device.h | 34 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| renamed_device.h | 85 Status MakeTensorFromProto(const TensorProto& tensor_proto,
88 return underlying_->MakeTensorFromProto(tensor_proto, alloc_attrs, tensor);
|
| constant_folding_test.cc | 50 const TensorProto* tensor_proto; local
51 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor_proto));
55 EXPECT_TRUE(t.FromProto(*tensor_proto));
63 const TensorProto* tensor_proto; local
64 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor_proto));
68 EXPECT_TRUE(t.FromProto(*tensor_proto));
|
| /external/tensorflow/tensorflow/contrib/util/ |
| convert_graphdef_memmapped_format_lib.cc | 45 const TensorProto& tensor_proto = value.tensor(); local
50 const DataType tensor_data_type = tensor_proto.dtype();
51 const TensorShapeProto tensor_shape = tensor_proto.tensor_shape();
68 Tensor parsed(tensor_proto.dtype());
69 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) {
71 tensor_proto.DebugString());
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| sycl_device.cc | 46 Status SYCLDevice::MakeTensorFromProto(const TensorProto& tensor_proto,
53 Tensor parsed(tensor_proto.dtype());
54 if (!parsed.FromProto(host_alloc, tensor_proto)) {
56 tensor_proto.DebugString());
|
| sycl_device.h | 215 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compilation_device.h | 62 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| xla_compilation_device.cc | 121 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs,
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_worker.cc | 101 TensorProto* tensor_proto = proto->mutable_tensor();
102 tensor_proto->set_dtype(val.dtype());
103 val.shape().AsProto(tensor_proto->mutable_tensor_shape());
|
| /external/tensorflow/tensorflow/core/debug/ |
| debug_grpc_testlib.cc | 53 const TensorProto& tensor_proto = val.tensor(); local
54 Tensor tensor(tensor_proto.dtype());
55 if (!tensor.FromProto(tensor_proto)) {
|
| debug_io_utils.cc | 126 TensorProto* tensor_proto,
128 const protobuf::RepeatedPtrField<string>& strs = tensor_proto->string_val();
157 num_chunks, i, tensor_proto->dtype(),
158 tensor_proto->tensor_shape());
163 tensor_proto->mutable_string_val());
187 TensorProto tensor_proto; local
192 tensor.AsProtoField(&tensor_proto);
195 debug_node_key, wall_time_us, chunk_size_limit, &tensor_proto, events));
197 tensor.AsProtoTensorContent(&tensor_proto);
199 const size_t total_length = tensor_proto.tensor_content().size()
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| ops.h | 183 TensorProto tensor_proto; local
185 tensor.AsProtoTensorContent(&tensor_proto);
187 tensor.AsProtoField(&tensor_proto);
189 return tensor_proto;
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| xla_device.cc | 258 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto,
263 Tensor parsed(tensor_proto.dtype());
264 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) {
266 tensor_proto.DebugString());
|
| xla_device.h | 94 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_utils.h | 86 TensorProto tensor_proto; local
87 tensor.AsProtoTensorContent(&tensor_proto);
88 SetNodeAttr(key, tensor_proto, node);
|
| /external/tensorflow/tensorflow/core/framework/ |
| device_base.h | 218 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| debug_data.py | 61 def __init__(self, tensor_proto, initialized=True):
65 tensor_proto: the `TensorProto` object that cannot be represented as a
69 self._tensor_proto = tensor_proto
120 tensor_proto = event.summary.value[0].tensor
121 shape = tensor_util.TensorShapeProtoToList(tensor_proto.tensor_shape)
126 if tensor_proto.tensor_content or tensor_proto.string_val or not num_elements:
128 if tensor_proto.dtype == types_pb2.DT_RESOURCE:
129 tensor_value = InconvertibleTensorProto(tensor_proto)
132 tensor_value = tensor_util.MakeNdarray(tensor_proto)
[all...] |
| /external/tensorflow/tensorflow/python/debug/cli/ |
| tensor_format_test.py | 342 tensor_proto = tensor_pb2.TensorProto(
347 debug_data.InconvertibleTensorProto(tensor_proto, False), "a")
351 self.assertEqual(str(tensor_proto).split("\n"), out.lines[3:])
354 tensor_proto = tensor_pb2.TensorProto(
359 debug_data.InconvertibleTensorProto(tensor_proto), "a")
362 self.assertEqual(str(tensor_proto).split("\n"), out.lines[2:])
626 tensor_proto = tensor_pb2.TensorProto(
631 debug_data.InconvertibleTensorProto(tensor_proto, False), "a")
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| gpu_device.h | 80 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.h | 206 static Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| nccl_rewrite.cc | 201 TensorProto tensor_proto = TensorFromShape(shape_proto); local
222 shape_builder.Attr("value", tensor_proto).Attr("dtype", DT_INT32);
|
