Home
last modified time | relevance | path

Searched refs:tensor_proto (Results 1 – 25 of 65) sorted by relevance

123

/external/tensorflow/tensorflow/python/framework/
Dfast_tensor_util.pyx10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
14 tensor_proto.half_val.append(nparray[i])
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
26 tensor_proto.half_val.append(nparray[i])
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
34 tensor_proto.float_val.append(nparray[i])
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
42 tensor_proto.double_val.append(nparray[i])
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
50 tensor_proto.int_val.append(nparray[i])
[all …]
Dtensor_util.py50 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
51 tensor_proto.half_val.extend(
55 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
58 tensor_proto,
67 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
68 tensor_proto.half_val.extend(
72 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
74 tensor_proto, np.asarray(
126 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
127 tensor_proto.float_val.extend([x.item() for x in proto_values])
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dunique_op_test.cc41 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
42 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProto()
43 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim); in GetRandomInt32TensorProto()
44 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false); in GetRandomInt32TensorProto()
47 tensor_proto.add_int_val(int_val); in GetRandomInt32TensorProto()
49 return tensor_proto; in GetRandomInt32TensorProto()
54 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
55 tensor_proto.set_dtype(DT_INT32); in GetRandomInt32TensorProtoWithRepeat()
56 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(dim); in GetRandomInt32TensorProtoWithRepeat()
57 tensor_proto.mutable_tensor_shape()->set_unknown_rank(false); in GetRandomInt32TensorProtoWithRepeat()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dtensor_util_test.cc438 TensorProto tensor_proto = in TEST() local
440 EXPECT_FALSE(tensor::CompressTensorProtoInPlace(&tensor_proto)); in TEST()
441 tensor_proto = in TEST()
443 EXPECT_FALSE(tensor::CompressTensorProtoInPlace(&tensor_proto)); in TEST()
444 tensor_proto = in TEST()
446 EXPECT_FALSE(tensor::CompressTensorProtoInPlace(&tensor_proto)); in TEST()
447 tensor_proto = in TEST()
449 EXPECT_FALSE(tensor::CompressTensorProtoInPlace(&tensor_proto)); in TEST()
450 tensor_proto = in TEST()
452 EXPECT_FALSE(tensor::CompressTensorProtoInPlace(&tensor_proto)); in TEST()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device.cc89 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto() argument
91 if (tensor_proto.dtype() > 0 && tensor_proto.dtype() <= DataType_MAX) { in MakeTensorFromProto()
92 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
93 if (parsed.FromProto(allocator_, tensor_proto)) { in MakeTensorFromProto()
99 tensor_proto.DebugString()); in MakeTensorFromProto()
Dsingle_threaded_cpu_device.cc61 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() argument
64 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
65 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
Drenamed_device.h106 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() argument
109 return underlying_device_->MakeTensorFromProto(tensor_proto, alloc_attrs, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() argument
82 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
84 if (!parsed.FromProto(allocator, tensor_proto)) { in MakeTensorFromProto()
86 tensor_proto.DebugString()); in MakeTensorFromProto()
Dvirtual_device.h31 Status MakeTensorFromProto(const TensorProto& tensor_proto,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor.cc63 TensorProto tensor_proto; in ConvertToProto() local
71 input_tensor.AsProtoTensorContent(&tensor_proto); in ConvertToProto()
73 input_tensor.AsProtoField(&tensor_proto); in ConvertToProto()
74 return tensor_proto; in ConvertToProto()
376 TensorProto tensor_proto; in ConvertToTensor() local
377 TF_RETURN_IF_ERROR(ConvertToTensorProto(attr, &tensor_proto)); in ConvertToTensor()
378 if (!output_tensor->FromProto(tensor_proto)) { in ConvertToTensor()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc50 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() argument
53 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
54 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc475 const TensorProto& tensor_proto, in MakeTensorFromProto() argument
478 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
479 if (!parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
481 tensor_proto.DebugString()); in MakeTensorFromProto()
499 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto() argument
508 return MakeTensorFromProto(device_contexts.first, tensor_proto, alloc_attrs, in MakeTensorFromProto()
513 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeFastMemTensorFromProto() argument
521 return MakeTensorFromProto(device_contexts.second, tensor_proto, alloc_attrs, in MakeFastMemTensorFromProto()
Dxla_device.h155 Status MakeTensorFromProto(const TensorProto& tensor_proto,
162 Status MakeFastMemTensorFromProto(const TensorProto& tensor_proto,
206 const TensorProto& tensor_proto,
/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils.cc129 TensorProto* tensor_proto, in WrapStringTensorAsEvents() argument
131 const protobuf::RepeatedPtrField<string>& strs = tensor_proto->string_val(); in WrapStringTensorAsEvents()
160 num_chunks, i, tensor_proto->dtype(), in WrapStringTensorAsEvents()
161 tensor_proto->tensor_shape()); in WrapStringTensorAsEvents()
166 tensor_proto->mutable_string_val()); in WrapStringTensorAsEvents()
190 TensorProto tensor_proto; in WrapTensorAsEvents() local
195 tensor.AsProtoField(&tensor_proto); in WrapTensorAsEvents()
198 debug_node_key, wall_time_us, chunk_size_limit, &tensor_proto, events)); in WrapTensorAsEvents()
200 tensor.AsProtoTensorContent(&tensor_proto); in WrapTensorAsEvents()
202 const size_t total_length = tensor_proto.tensor_content().size(); in WrapTensorAsEvents()
[all …]
Ddebug_grpc_testlib.cc54 const TensorProto& tensor_proto = val.tensor(); in SendEvents() local
55 Tensor tensor(tensor_proto.dtype()); in SendEvents()
56 if (!tensor.FromProto(tensor_proto)) { in SendEvents()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc398 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
399 tensor_proto.set_dtype(dtype); in MakeTensorProtoFromShape()
400 auto* shape_proto = tensor_proto.mutable_tensor_shape(); in MakeTensorProtoFromShape()
408 tensor_proto.add_int_val(value); in MakeTensorProtoFromShape()
410 tensor_proto.add_int64_val(value); in MakeTensorProtoFromShape()
413 return tensor_proto; in MakeTensorProtoFromShape()
418 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto() argument
426 *tensor = tensor_proto; in MakeConstNodeDefFromTensorProto()
1004 const auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
1005 if (tensor_proto != nullptr) { in UpdateNode()
[all …]
Dop_level_cost_estimator_test.cc398 TensorProto* tensor_proto) { in GetTensorProto() argument
399 tensor_proto->Clear(); in GetTensorProto()
422 tensor.AsProtoTensorContent(tensor_proto); in GetTensorProto()
424 tensor.AsProtoField(tensor_proto); in GetTensorProto()
446 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp() local
448 /*tensor_content=*/false, tensor_proto); in DescribePoolingOp()
1313 TensorProto tensor_proto; in TEST_F() local
1318 tensor_proto.mutable_tensor_shape()->add_dim()->set_size(255); in TEST_F()
1320 GetTensorShapeProtoFromTensorProto(tensor_proto, &tensor_shape_proto)); in TEST_F()
1322 tensor_proto.Clear(); in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Ddebug_data.py62 def __init__(self, tensor_proto, initialized=True): argument
70 self._tensor_proto = tensor_proto
121 tensor_proto = event.summary.value[0].tensor
122 shape = tensor_util.TensorShapeProtoToList(tensor_proto.tensor_shape)
127 if tensor_proto.tensor_content or tensor_proto.string_val or not num_elements:
129 if tensor_proto.dtype == types_pb2.DT_RESOURCE:
130 tensor_value = InconvertibleTensorProto(tensor_proto)
133 tensor_value = tensor_util.MakeNdarray(tensor_proto)
135 tensor_value = InconvertibleTensorProto(tensor_proto)
138 tensor_value = InconvertibleTensorProto(tensor_proto, False)
Ddumping_callback.py518 tensor_proto = _concrete_tensor_to_proto(
525 tensor_proto = tensor_pb2.TensorProto()
529 tensor_proto = _concrete_tensor_to_proto(
536 tensor_proto = tensor_pb2.TensorProto()
538 tensor_proto = _concrete_tensor_to_proto(tensor)
539 if tensor_proto:
540 execution_proto.tensor_protos.append(tensor_proto)
Ddebug_events_reader.py845 def _parse_tensor_value(tensor_proto, return_list=False): argument
873 ndarray = tensor_util.MakeNdarray(tensor_proto)
902 for tensor_proto in execution_proto.tensor_protos:
905 _parse_tensor_value(tensor_proto, return_list=True))
1080 trace_proto.tensor_proto, return_list=True)
1323 return [_parse_tensor_value(tensor_proto)
1324 for tensor_proto in debug_event.execution.tensor_protos]
1337 return _parse_tensor_value(debug_event.graph_execution_trace.tensor_proto)
/external/tensorflow/tensorflow/cc/framework/
Dops.h183 TensorProto tensor_proto;
185 tensor.AsProtoTensorContent(&tensor_proto);
187 tensor.AsProtoField(&tensor_proto);
189 return tensor_proto;
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h86 TensorProto tensor_proto; in SetNodeTensorAttr() local
87 tensor.AsProtoTensorContent(&tensor_proto); in SetNodeTensorAttr()
88 SetNodeAttr(key, tensor_proto, node); in SetNodeTensorAttr()
/external/tensorflow/tensorflow/python/debug/cli/
Dtensor_format_test.py342 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:])
613 tensor_proto = tensor_pb2.TensorProto(
618 debug_data.InconvertibleTensorProto(tensor_proto, False), "a")
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dcluster_function_library_runtime.cc241 for (const auto& tensor_proto : response->tensor()) { in Run() local
243 if (t.FromProto(tensor_proto)) { in Run()
247 tensor_proto.DebugString())); in Run()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc737 const TensorProto* tensor_proto = nullptr; in RegisterPadNode() local
738 TF_CHECK_OK(GetNodeAttr(input_node->attrs(), "value", &tensor_proto)); in RegisterPadNode()
739 CHECK_NOTNULL(tensor_proto); in RegisterPadNode()
741 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor)); in RegisterPadNode()
1101 const TensorProto& tensor_proto, Tensor* tensor) { in MakeTensorFromProto() argument
1102 if (tensor_proto.dtype() > 0 && tensor_proto.dtype() <= DataType_MAX) { in MakeTensorFromProto()
1103 Tensor parsed(tensor_proto.dtype()); in MakeTensorFromProto()
1104 if (parsed.FromProto(cpu_allocator(), tensor_proto)) { in MakeTensorFromProto()
1110 tensor_proto.DebugString()); in MakeTensorFromProto()

123