/external/tensorflow/tensorflow/python/framework/ |
D | fast_tensor_util.pyx | 10 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 …]
|
D | tensor_util.py | 50 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/ |
D | unique_op_test.cc | 41 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/ |
D | tensor_util_test.cc | 438 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/ |
D | threadpool_device.cc | 89 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()
|
D | single_threaded_cpu_device.cc | 61 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()
|
D | renamed_device.h | 106 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/ |
D | virtual_device.cc | 79 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()
|
D | virtual_device.h | 31 Status MakeTensorFromProto(const TensorProto& tensor_proto,
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | convert_tensor.cc | 63 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/ |
D | evaluation_utils.cc | 50 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/ |
D | xla_device.cc | 475 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()
|
D | xla_device.h | 155 Status MakeTensorFromProto(const TensorProto& tensor_proto, 162 Status MakeFastMemTensorFromProto(const TensorProto& tensor_proto, 206 const TensorProto& tensor_proto,
|
/external/tensorflow/tensorflow/core/debug/ |
D | debug_io_utils.cc | 129 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 …]
|
D | debug_grpc_testlib.cc | 54 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/ |
D | graph_properties.cc | 398 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 …]
|
D | op_level_cost_estimator_test.cc | 398 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/ |
D | debug_data.py | 62 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)
|
D | dumping_callback.py | 518 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)
|
D | debug_events_reader.py | 845 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/ |
D | ops.h | 183 TensorProto tensor_proto; 185 tensor.AsProtoTensorContent(&tensor_proto); 187 tensor.AsProtoField(&tensor_proto); 189 return tensor_proto;
|
/external/tensorflow/tensorflow/tools/graph_transforms/ |
D | transform_utils.h | 86 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/ |
D | 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:]) 613 tensor_proto = tensor_pb2.TensorProto( 618 debug_data.InconvertibleTensorProto(tensor_proto, False), "a")
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | cluster_function_library_runtime.cc | 241 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/ |
D | graph_transferer.cc | 737 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()
|