Home
last modified time | relevance | path

Searched defs:tensor_proto (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/python/framework/
Dfast_tensor_util.pyx10 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
22 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
30 tensor_proto, np.ndarray[np.float32_t, ndim=1] nparray): argument
38 tensor_proto, np.ndarray[np.float64_t, ndim=1] nparray): argument
46 tensor_proto, np.ndarray[np.int32_t, ndim=1] nparray): argument
53 tensor_proto, np.ndarray[np.uint32_t, ndim=1] nparray): argument
60 tensor_proto, np.ndarray[np.int64_t, ndim=1] nparray): argument
67 tensor_proto, np.ndarray[np.uint64_t, ndim=1] nparray): argument
74 tensor_proto, np.ndarray[np.uint8_t, ndim=1] nparray): argument
82 tensor_proto, np.ndarray[np.uint16_t, ndim=1] nparray): argument
[all …]
Dtensor_util.py49 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
54 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
66 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
71 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
125 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
128 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values): argument
131 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values): argument
134 def SlowAppendInt64ArrayToTensorProto(tensor_proto, proto_values): argument
137 def SlowAppendQIntArrayToTensorProto(tensor_proto, proto_values): argument
140 def SlowAppendUInt32ArrayToTensorProto(tensor_proto, proto_values): argument
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dunique_op_test.cc39 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
52 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
106 TensorProto tensor_proto; in GetRandomStringsTensorProto() local
/external/tensorflow/tensorflow/core/framework/
Dtensor_util_test.cc437 TensorProto tensor_proto = in TEST() local
459 TensorProto tensor_proto = in TEST() local
512 TensorProto tensor_proto; in CreateAsProtoTensorContent() local
522 TensorProto tensor_proto; in CreateAsProtoField() local
542 TensorProto tensor_proto = in ConstantTailTest() local
Ddevice_base.h243 virtual Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/common_runtime/sycl/
Dsycl_device.cc45 Status SYCLDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/external/tensorflow/tensorflow/core/common_runtime/
Dsingle_threaded_cpu_device.cc65 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
Dthreadpool_device.cc92 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
Drenamed_device.h91 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
Dconstant_folding_test.cc54 const TensorProto* tensor_proto; in ExpectNodeClose() local
67 const TensorProto* tensor_proto; in ExpectNodeEqual() local
/external/tensorflow/tensorflow/core/debug/
Ddebug_grpc_testlib.cc54 const TensorProto& tensor_proto = val.tensor(); in SendEvents() local
Ddebug_io_utils.cc129 TensorProto* tensor_proto, in WrapStringTensorAsEvents()
190 TensorProto tensor_proto; in WrapTensorAsEvents() local
/external/tensorflow/tensorflow/contrib/util/
Dconvert_graphdef_memmapped_format_lib.cc45 const TensorProto& tensor_proto = value.tensor(); in ConvertConstantsToImmutable() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.cc71 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.cc119 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc98 TensorProto* tensor_proto = proto->mutable_tensor(); in GrpcRecvTensorAsync() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h86 TensorProto tensor_proto; in SetNodeTensorAttr() local
Dtransform_utils_test.cc186 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttr() local
202 TensorProto tensor_proto = node.attr().at("foo").tensor(); in TestSetNodeTensorAttrWithTensor() local
212 TensorProto tensor_proto; in TestGetNodeTensorAttr() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc372 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
392 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto()
791 auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
1315 TensorProto tensor_proto; in UpdateOutputShapesAndValues() local
1381 const TensorProto& tensor_proto = node.attr().at("value").tensor(); in MaybeUpdateNodeContextOutput() local
1628 TensorProto tensor_proto; in MakeIntegerScalarTensorProto() local
1641 const TensorProto& tensor_proto, in MaybeTensorProtoToShape()
Dop_level_cost_estimator_test.cc338 TensorProto* tensor_proto) { in GetTensorProto()
386 auto* tensor_proto = op_info.mutable_inputs(0)->mutable_value(); in DescribePoolingOp() local
936 TensorProto tensor_proto; in TEST_F() local
Dutils.cc66 for (const auto& tensor_proto : attr_value.list().tensor()) { in ExtractTensors() local
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc203 TensorProto tensor_proto = TensorFromShape(shape_proto); in ReplaceBroadcast() local
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc282 for (const auto& tensor_proto : request->tensors()) { in SendTensor() local
Deager_service_impl_test.cc84 void SetTensorProto(TensorProto* tensor_proto) { in SetTensorProto()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc440 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()

12