| /external/tensorflow/tensorflow/python/framework/ |
| D | fast_tensor_util.pyx | 10 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 …]
|
| D | tensor_util.py | 46 def SlowAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
51 def _MediumAppendFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
63 def SlowAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
68 def FastAppendBFloat16ArrayToTensorProto(tensor_proto, proto_values): argument
122 def SlowAppendFloat32ArrayToTensorProto(tensor_proto, proto_values): argument
125 def SlowAppendFloat64ArrayToTensorProto(tensor_proto, proto_values): argument
128 def SlowAppendIntArrayToTensorProto(tensor_proto, proto_values): argument
131 def SlowAppendInt64ArrayToTensorProto(tensor_proto, proto_values): argument
134 def SlowAppendQIntArrayToTensorProto(tensor_proto, proto_values): argument
137 def SlowAppendUInt32ArrayToTensorProto(tensor_proto, proto_values): argument
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | unique_op_test.cc | 41 TensorProto tensor_proto; in GetRandomInt32TensorProto() local
54 TensorProto tensor_proto; in GetRandomInt32TensorProtoWithRepeat() local
115 TensorProto tensor_proto; in GetRandomStringsTensorProto() local
|
| D | control_flow_ops_test.cc | 182 TensorProto tensor_proto; in TEST() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util_test.cc | 463 TensorProto tensor_proto = in TEST() local
485 TensorProto tensor_proto = in TEST() local
551 TensorProto tensor_proto; in CreateAsProtoTensorContent() local
562 TensorProto tensor_proto; in CreateAsProtoField() local
582 TensorProto tensor_proto = in ConstantTailTest() local
635 TensorProto tensor_proto; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | auto_shard_dataset_op_test.cc | 249 TensorProto tensor_proto; in TEST_F() local
305 TensorProto tensor_proto; in TEST_F() local
356 TensorProto tensor_proto; in TEST_F() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 61 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| D | renamed_device.h | 107 Status MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| D | threadpool_device.cc | 122 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.cc | 50 Status DeviceSimple::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 79 Status VirtualDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 66 const tensorflow::TensorProto &tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor.cc | 62 TensorProto tensor_proto; in ConvertToProto() local
482 TensorProto tensor_proto; in ConvertToTensor() local
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | debug_grpc_testlib.cc | 54 const TensorProto& tensor_proto = val.tensor(); in SendEvents() local
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_tensor.cc | 59 TensorProto tensor_proto; in ConvertToProto() local
481 TensorProto tensor_proto; in ConvertToTensor() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 147 const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | convert_tf_quant_ops_to_mhlo.cc | 131 tensorflow::TensorProto tensor_proto; in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | small_constant_optimization.cc | 104 TensorProto tensor_proto; in ExtractSmallTensorValue() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 402 TensorProto tensor_proto; in MakeTensorProtoFromShape() local
422 const TensorProto& tensor_proto, in MakeConstNodeDefFromTensorProto()
475 uint64 NumElementsFromTensorProto(const TensorProto& tensor_proto) { in NumElementsFromTensorProto()
1033 const auto* tensor_proto = src_ctx->output_tensor_protos[src_output]; in UpdateNode() local
1117 const TensorProto* tensor_proto = ctx->input_tensor_protos[dst_input]; in UpdateNode() local
1625 TensorProto tensor_proto; in UpdateOutputShapesAndValues() local
1714 const TensorProto& tensor_proto = node.attr().at("value").tensor(); in MaybeUpdateNodeContextOutput() local
1967 TensorProto tensor_proto; in MakeIntegerScalarTensorProto() local
1980 const TensorProto& tensor_proto, in MaybeTensorProtoToShape()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 505 const TensorProto& tensor_proto, in MakeTensorFromProto()
532 Status XlaDevice::MakeTensorFromProto(const TensorProto& tensor_proto, in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.h | 86 TensorProto tensor_proto; in SetNodeTensorAttr() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | batch_dataset_op_test.cc | 350 TensorProto tensor_proto; in TEST() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | debug_event.proto | 296 TensorProto tensor_proto = 5; field
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 203 TensorProto tensor_proto = TensorFromShape(shape_proto); in ReplaceBroadcast() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | cluster_function_library_runtime.cc | 242 for (const auto& tensor_proto : response->tensor()) { in Run() local
|