| /external/protobuf/src/google/protobuf/stubs/ |
| D | statusor_test.cc | 71 TEST(StatusOr, TestDefaultCtor) { in TEST() argument
72 StatusOr<int> thing; in TEST()
77 TEST(StatusOr, TestStatusCtor) { in TEST() argument
78 StatusOr<int> thing(Status::CANCELLED); in TEST()
83 TEST(StatusOr, TestValueCtor) { in TEST() argument
85 StatusOr<int> thing(kI); in TEST()
90 TEST(StatusOr, TestCopyCtorStatusOk) { in TEST() argument
92 StatusOr<int> original(kI); in TEST()
93 StatusOr<int> copy(original); in TEST()
98 TEST(StatusOr, TestCopyCtorStatusNotOk) { in TEST() argument
[all …]
|
| D | statusor.h | 97 class StatusOr {
98 template<typename U> friend class StatusOr; variable
102 StatusOr();
114 StatusOr(const Status& status); // NOLINT
127 StatusOr(const T& value); // NOLINT
130 StatusOr(const StatusOr& other);
134 StatusOr(const StatusOr<U>& other);
137 StatusOr& operator=(const StatusOr& other);
141 StatusOr& operator=(const StatusOr<U>& other);
189 inline StatusOr<T>::StatusOr()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/lib/ |
| D | statusor.h | 81 class TF_MUST_USE_RESULT StatusOr; variable
85 class StatusOr : private internal_statusor::StatusOrData<T>,
90 friend class StatusOr; variable
101 explicit StatusOr();
105 StatusOr(const StatusOr&) = default;
106 StatusOr& operator=(const StatusOr&) = default;
110 StatusOr(StatusOr&&) = default;
111 StatusOr& operator=(StatusOr&&) = default;
116 StatusOr(const StatusOr<U>& other);
119 StatusOr(StatusOr<U>&& other);
[all …]
|
| D | statusor_test.cc | 68 StatusOr<std::unique_ptr<int>> ReturnUniquePtr() { in ReturnUniquePtr()
73 TEST(StatusOr, ElementType) { in TEST() argument
74 static_assert(std::is_same<StatusOr<int>::element_type, int>(), ""); in TEST()
75 static_assert(std::is_same<StatusOr<char>::element_type, char>(), ""); in TEST()
78 TEST(StatusOr, NullPointerStatusOr) { in TEST() argument
81 StatusOr<int*> null_status(nullptr); in TEST()
86 TEST(StatusOr, TestNoDefaultConstructorInitialization) { in TEST() argument
88 StatusOr<NoDefaultConstructor> statusor(tensorflow::errors::Cancelled("")); in TEST()
93 StatusOr<NoDefaultConstructor> statusor2; in TEST()
98 TEST(StatusOr, TestMoveOnlyInitialization) { in TEST() argument
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/platform/default/ |
| D | dso_loader.h | 37 port::StatusOr<void*> GetCudaDriverDsoHandle();
38 port::StatusOr<void*> GetCudaRuntimeDsoHandle();
39 port::StatusOr<void*> GetCublasDsoHandle();
40 port::StatusOr<void*> GetCufftDsoHandle();
41 port::StatusOr<void*> GetCurandDsoHandle();
42 port::StatusOr<void*> GetCuptiDsoHandle();
43 port::StatusOr<void*> GetCudnnDsoHandle();
45 port::StatusOr<void*> GetRocblasDsoHandle();
46 port::StatusOr<void*> GetMiopenDsoHandle();
47 port::StatusOr<void*> GetRocfftDsoHandle();
[all …]
|
| D | dso_loader.cc | 36 port::StatusOr<void*> GetDsoHandle(const string& name, const string& version) { in GetDsoHandle()
59 port::StatusOr<void*> GetCudaDriverDsoHandle() { in GetCudaDriverDsoHandle()
73 port::StatusOr<void*> GetCudaRuntimeDsoHandle() { in GetCudaRuntimeDsoHandle()
77 port::StatusOr<void*> GetCublasDsoHandle() { in GetCublasDsoHandle()
81 port::StatusOr<void*> GetCufftDsoHandle() { in GetCufftDsoHandle()
85 port::StatusOr<void*> GetCurandDsoHandle() { in GetCurandDsoHandle()
89 port::StatusOr<void*> GetCuptiDsoHandle() { in GetCuptiDsoHandle()
99 port::StatusOr<void*> GetCudnnDsoHandle() { in GetCudnnDsoHandle()
103 port::StatusOr<void*> GetRocblasDsoHandle() { in GetRocblasDsoHandle()
107 port::StatusOr<void*> GetMiopenDsoHandle() { in GetMiopenDsoHandle()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.h | 49 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
51 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
56 static StatusOr<Shape> InferBinaryOpShape(
59 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
65 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode, const Shape& lhs,
68 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode,
75 static StatusOr<Shape> InferVariadicOpShape(
77 static StatusOr<Shape> InferVariadicOpShape(
82 static StatusOr<Shape> InferMapShape(
88 static StatusOr<Shape> InferBatchNormTrainingShape(const Shape& operand_shape,
[all …]
|
| D | hlo_creation_utils.h | 32 StatusOr<HloInstruction*> MakeBinaryHlo(HloOpcode opcode, HloInstruction* lhs,
37 StatusOr<HloInstruction*> MakeCompareHlo(ComparisonDirection direction,
44 StatusOr<HloInstruction*> MakePadHlo(HloInstruction* operand,
50 StatusOr<HloInstruction*> MakeSliceHlo(HloInstruction* operand,
57 StatusOr<HloInstruction*> MakeConvolveHlo(
64 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand,
69 StatusOr<HloInstruction*> MakeReshapeHlo(const Shape& result_shape,
72 StatusOr<HloInstruction*> MakeReshapeHlo(
78 StatusOr<HloInstruction*> MakeDynamicSliceHlo(
85 StatusOr<HloInstruction*> MakeDynamicUpdateSliceHlo(
[all …]
|
| D | elemental_ir_emitter.h | 43 virtual StatusOr<llvm::Value*> EmitUnaryOp(const HloInstruction* op,
46 virtual StatusOr<llvm::Value*> EmitBinaryOp(const HloInstruction* op,
64 virtual StatusOr<llvm::Value*> EmitIntegerUnaryOp(const HloInstruction* op,
67 virtual StatusOr<llvm::Value*> EmitFloatUnaryOp(const HloInstruction* op,
70 virtual StatusOr<llvm::Value*> EmitComplexUnaryOp(const HloInstruction* op,
85 virtual StatusOr<llvm::Value*> EmitIntegerBinaryOp(const HloInstruction* op,
90 virtual StatusOr<llvm::Value*> EmitFloatBinaryOp(const HloInstruction* op,
94 virtual StatusOr<llvm::Value*> EmitComplexBinaryOp(const HloInstruction* op,
110 virtual StatusOr<llvm::Value*> EmitErfInv(PrimitiveType prim_type,
113 virtual StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
[all …]
|
| D | hlo_runner.h | 89 static StatusOr<std::unique_ptr<HloModule>> CreateModuleFromString(
94 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromBinaryProtoFile(
96 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromTextProtoFile(
101 static StatusOr<std::unique_ptr<HloModule>> ReadModuleFromHloTextFile(
105 StatusOr<ScopedShapedBuffer> TransferLiteralToDevice(const Literal& literal);
106 StatusOr<std::vector<ScopedShapedBuffer>> TransferLiteralsToDevice(
108 StatusOr<std::vector<ScopedShapedBuffer>> TransferLiteralsToDevice(
110 StatusOr<Literal> TransferLiteralFromDevice(const ShapedBuffer& buffer);
117 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
122 StatusOr<Literal> Execute(std::unique_ptr<HloModule> module,
[all …]
|
| D | hlo_parser.h | 35 StatusOr<std::unique_ptr<HloModule>> ParseHloString(
45 StatusOr<std::unique_ptr<HloModule>> ParseHloString(absl::string_view str);
50 StatusOr<HloSharding> ParseSharding(absl::string_view str);
55 StatusOr<std::vector<bool>> ParseParameterReplication(absl::string_view str);
58 StatusOr<Window> ParseWindow(absl::string_view str);
62 StatusOr<ConvolutionDimensionNumbers> ParseConvolutionDimensionNumbers(
66 StatusOr<PaddingConfig> ParsePaddingConfig(absl::string_view str);
69 StatusOr<Shape> ParseShape(absl::string_view str);
|
| D | hlo_memory_scheduler.h | 38 typedef std::function<StatusOr<HloInstructionSequence>(
45 StatusOr<HloInstructionSequence> ListMemoryScheduler(
53 StatusOr<HloInstructionSequence> DFSMemoryScheduler(
61 StatusOr<HloInstructionSequence> PostOrderMemoryScheduler(
71 StatusOr<HloInstructionSequence> DefaultMemoryScheduler(
81 StatusOr<HloSchedule> ScheduleModule(
87 StatusOr<HloInstructionSequence> ScheduleComputation(
104 StatusOr<bool> Run(HloModule* module) override;
117 StatusOr<bool> Run(HloModule* module) override;
128 StatusOr<bool> Run(HloModule* module) override;
|
| D | indexed_array_analysis.h | 248 StatusOr<Array*> GetArrayFor(const HloInstruction* instr);
260 StatusOr<Array*> ComputeArrayFor(const HloInstruction* instr);
262 StatusOr<Array*> ComputeArrayForConstant(const Literal& literal);
264 StatusOr<Array*> ComputeArrayForGather(
268 StatusOr<Array*> ComputeArrayForDotWithIndexedLhs(
273 StatusOr<Array*> ComputeArrayForDotWithIndexedRhs(
278 StatusOr<Array*> ComputeArrayForDot(const Shape& shape,
304 StatusOr<ScalarIndexedArray*> FoldGatherOfGather(
310 StatusOr<ScalarIndexedArray*> ReshapeToRemoveDegenerateDims(
315 StatusOr<ScalarIndexedArray*> ReshapeToAddDegenerateDims(
[all …]
|
| D | service.h | 85 static StatusOr<std::unique_ptr<Service>> NewService(
87 static StatusOr<std::unique_ptr<Service>> NewService(
189 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
195 StatusOr<std::vector<se::StreamExecutor*>> GetExecutors(
200 StatusOr<std::vector<std::vector<const ShapedBuffer*>>> GetArguments(
216 StatusOr<std::vector<std::vector<const ShapedBuffer*>>>
223 StatusOr<std::unique_ptr<HloModuleConfig>> CreateModuleConfig(
233 StatusOr<std::unique_ptr<Executable>> BuildExecutable(
241 StatusOr<std::vector<std::unique_ptr<Executable>>> BuildExecutables(
251 StatusOr<GlobalDataHandle> ExecuteAndRegisterResult(
[all …]
|
| D | platform_util.h | 38 static StatusOr<std::vector<se::Platform*>> GetSupportedPlatforms();
45 static StatusOr<se::Platform*> GetDefaultPlatform();
50 static StatusOr<se::Platform*> GetSolePlatform();
54 static StatusOr<se::Platform*> GetPlatform(const string& platform_name);
58 static StatusOr<se::Platform*> GetPlatformExceptFor(
69 static StatusOr<std::vector<se::StreamExecutor*>> GetStreamExecutors(
|
| D | allocation_tracker.h | 49 StatusOr<GlobalDataHandle> Register(ScopedShapedBuffer shaped_buffer,
54 StatusOr<GlobalDataHandle> RegisterReplicatedBuffers(
61 StatusOr<std::vector<GlobalDataHandle>> DeconstructTuple(
67 StatusOr<std::vector<const ShapedBuffer*>> Resolve(
73 StatusOr<const ShapedBuffer*> ResolveForReplica(const GlobalDataHandle& data,
89 StatusOr<std::vector<const ShapedBuffer*>> ResolveInternal(
97 StatusOr<GlobalDataHandle> RegisterInternal(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client.h | 61 StatusOr<ExecutionHandle> Compile(
70 StatusOr<std::unique_ptr<GlobalData>> Execute(
89 StatusOr<std::unique_ptr<GlobalData>> Execute(
119 StatusOr<std::vector<std::unique_ptr<GlobalData>>> ExecuteParallel(
126 StatusOr<std::vector<DeviceHandle>> GetDeviceHandles(int64 device_count);
134 StatusOr<Literal> Transfer(const GlobalData& data,
144 StatusOr<std::unique_ptr<GlobalData>> TransferToServer(
160 StatusOr<Literal> TransferFromOutfeed(
170 StatusOr<Literal> ExecuteAndTransfer(
191 StatusOr<Literal> ComputeConstant(
[all …]
|
| D | local_client.h | 42 StatusOr<ScopedShapedBuffer> Run(
74 StatusOr<ScopedShapedBuffer> ExecuteAndDump(
87 StatusOr<Literal> LiteralFromShapedBuffer(const ShapedBuffer& shaped_buffer);
119 StatusOr<std::unique_ptr<LocalExecutable>> Compile(
128 StatusOr<ScopedShapedBuffer> LiteralToShapedBuffer(
133 StatusOr<TransferToServerResponse> TransferToLocalServer(
138 StatusOr<Literal> ShapedBufferToLiteral(const ShapedBuffer& shaped_buffer);
142 StatusOr<const ShapedBuffer*> GlobalDataToShapedBuffer(
156 StatusOr<Literal> TransferFromOutfeedLocal(const Shape& shape,
164 StatusOr<int> ReplicaNumberToDeviceOrdinal(int replica_number);
|
| D | client.cc | 41 StatusOr<Literal> Client::Transfer(const GlobalData& data, in Transfer()
68 StatusOr<std::unique_ptr<GlobalData>> Client::TransferToServer( in TransferToServer()
118 StatusOr<Literal> Client::TransferFromOutfeed( in TransferFromOutfeed()
166 StatusOr<Literal> Client::ExecuteAndTransfer( in ExecuteAndTransfer()
184 StatusOr<Literal> Client::ComputeConstant(const XlaComputation& computation, in ComputeConstant()
212 StatusOr<XlaComputation> Client::LoadSnapshot(const HloSnapshot& module) { in LoadSnapshot()
217 StatusOr<ExecutionHandle> Client::Compile( in Compile()
251 StatusOr<std::unique_ptr<GlobalData>> Client::Execute( in Execute()
277 StatusOr<std::unique_ptr<GlobalData>> Client::Execute( in Execute()
328 StatusOr<std::vector<std::unique_ptr<GlobalData>>> Client::ExecuteParallel( in ExecuteParallel()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | elemental_ir_emitter.h | 42 using NestedComputer = std::function<StatusOr<llvm::Value*>(
54 StatusOr<llvm::Value*> EmitFloatBinaryOp(const HloInstruction* op,
58 StatusOr<llvm::Value*> EmitErfcInv(PrimitiveType prim_type,
61 StatusOr<llvm::Value*> EmitLog(PrimitiveType prim_type,
64 StatusOr<llvm::Value*> EmitLog1p(PrimitiveType prim_type,
67 StatusOr<llvm::Value*> EmitSin(PrimitiveType prim_type,
70 StatusOr<llvm::Value*> EmitCos(PrimitiveType prim_type,
73 StatusOr<llvm::Value*> EmitExp(PrimitiveType prim_type,
76 StatusOr<llvm::Value*> EmitExpm1(PrimitiveType prim_type,
79 StatusOr<llvm::Value*> EmitSqrt(PrimitiveType prim_type,
[all …]
|
| /external/protobuf/src/google/protobuf/util/internal/ |
| D | datapiece.h | 117 util::StatusOr<int32> ToInt32() const;
120 util::StatusOr<uint32> ToUint32() const;
123 util::StatusOr<int64> ToInt64() const;
126 util::StatusOr<uint64> ToUint64() const;
129 util::StatusOr<double> ToDouble() const;
132 util::StatusOr<float> ToFloat() const;
135 util::StatusOr<bool> ToBool() const;
138 util::StatusOr<string> ToString() const;
144 util::StatusOr<string> ToBytes() const;
152 util::StatusOr<int> ToEnum(const google::protobuf::Enum* enum_type) const;
[all …]
|
| D | datapiece.cc | 53 using util::StatusOr;
62 StatusOr<To> ValidateNumberConversion(To after, From before) { in ValidateNumberConversion()
79 StatusOr<To> NumberConvertAndCheck(From before) { in NumberConvertAndCheck()
89 StatusOr<To> FloatingPointToIntConvertAndCheck(From before) { in FloatingPointToIntConvertAndCheck()
98 StatusOr<To> FloatingPointConvertAndCheck(From before) { in FloatingPointConvertAndCheck()
115 StatusOr<int32> DataPiece::ToInt32() const { in ToInt32()
127 StatusOr<uint32> DataPiece::ToUint32() const { in ToUint32()
139 StatusOr<int64> DataPiece::ToInt64() const { in ToInt64()
151 StatusOr<uint64> DataPiece::ToUint64() const { in ToUint64()
163 StatusOr<double> DataPiece::ToDouble() const { in ToDouble()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | multi_platform_manager.cc | 35 port::StatusOr<Platform*> PlatformWithName(absl::string_view target)
38 port::StatusOr<Platform*> PlatformWithId(const Platform::Id& id)
41 port::StatusOr<Platform*> InitializePlatformWithName(
44 port::StatusOr<Platform*> InitializePlatformWithId(
57 port::StatusOr<Platform*> LookupByNameLocked(absl::string_view target)
62 port::StatusOr<Platform*> LookupByIdLocked(const Platform::Id& id)
95 port::StatusOr<Platform*> MultiPlatformManagerImpl::PlatformWithName( in PlatformWithName()
107 port::StatusOr<Platform*> MultiPlatformManagerImpl::PlatformWithId( in PlatformWithId()
119 port::StatusOr<Platform*> MultiPlatformManagerImpl::InitializePlatformWithName( in InitializePlatformWithName()
135 port::StatusOr<Platform*> MultiPlatformManagerImpl::InitializePlatformWithId( in InitializePlatformWithId()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 71 static port::StatusOr<GpuDeviceHandle> DeviceFromContext(GpuContext* context);
190 static port::StatusOr<GpuSharedMemConfig> ContextGetSharedMemConfig(
373 static port::StatusOr<GpuStatus> QueryEvent(GpuContext* context,
379 static port::StatusOr<GpuContext*> GetPointerContext(GpuDevicePtr pointer);
382 static port::StatusOr<GpuDeviceHandle> GetPointerDevice(GpuDevicePtr pointer);
385 static port::StatusOr<MemorySpace> GetPointerMemorySpace(
407 static port::StatusOr<int> GetMultiprocessorCount(GpuDeviceHandle device);
411 static port::StatusOr<int64> GetMaxThreadsPerMultiprocessor(
416 static port::StatusOr<int64> GetMaxThreadsPerBlock(GpuDeviceHandle device);
420 static port::StatusOr<int64> GetMaxSharedMemoryPerCore(
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_diagnostics.cc | 50 string DriverVersionStatusToString(port::StatusOr<DriverVersion> version) { in DriverVersionStatusToString()
58 port::StatusOr<DriverVersion> StringToDriverVersion(const string& value) { in StringToDriverVersion()
140 port::StatusOr<DriverVersion> dso_version = FindDsoVersion(); in LogDriverVersionInformation()
144 port::StatusOr<DriverVersion> kernel_version = FindKernelDriverVersion(); in LogDriverVersionInformation()
155 port::StatusOr<DriverVersion> Diagnostician::FindDsoVersion() { in FindDsoVersion()
156 port::StatusOr<DriverVersion> result{port::Status{ in FindDsoVersion()
183 auto result = static_cast<port::StatusOr<DriverVersion>*>(data); in FindDsoVersion()
195 port::StatusOr<DriverVersion> Diagnostician::FindKernelModuleVersion( in FindKernelModuleVersion()
218 port::StatusOr<DriverVersion> dso_version, in WarnOnDsoKernelMismatch()
219 port::StatusOr<DriverVersion> kernel_version) { in WarnOnDsoKernelMismatch()
[all …]
|