tensorflow/tensorflow/tensorflow/compiler/xla/ Source Tree

../
array.cc
array.h Array conjunction pack_is_integral
array2d.h Array2D
array3d.h Array3D
array4d.h Array4D
[+] client/
comparison_util.cc
comparison_util.h Comparison
cpu_function_runtime.cc
cpu_function_runtime.h BufferInfo
debug_options_flags.cc
debug_options_flags.h
debug_options_parsers.h
error_spec.h ErrorSpec
executable_run_options.cc
executable_run_options.h ExecutableRunOptions RunId
execution_options_util.cc
execution_options_util.h
index_util.cc
index_util.h IndexUtil
iterator_util.h FilteringIterator UnwrappingIterator
layout.cc
layout.h Layout Tile
layout_util.cc
layout_util.h LayoutUtil
literal.cc
literal.h BorrowingLiteral Literal LiteralBase LiteralSlice MutableBorrowingLiteral MutableLiteralBase
literal_comparison.cc
literal_comparison.h ErrorBuckets
literal_util.cc
literal_util.h LiteralUtil
map_util.h
metric_table_report.cc
metric_table_report.h MetricTableReport
[+] mlir/
[+] mlir_hlo/
overflow_util.h
parse_flags_from_env.cc
parse_flags_from_env.h
permutation_util.cc
permutation_util.h
[+] pjrt/
primitive_util.cc PrimitiveTypeNameGenerator
primitive_util.h PrimitiveTypeToNative
protobuf_util.cc
protobuf_util.h ProtobufEqualsWrapper ProtobufHashWrapper
refcounting_hash_map.h RefcountingHashMap
[+] runtime/
[+] service/
service_interface.h ServiceInterface
shape.cc
shape.h ProgramShape Shape
shape_layout.cc
shape_layout.h ShapeLayout
shape_tree.cc
shape_tree.h ShapeTree IndexTable
shape_util.cc
shape_util.h ShapeIndex ShapeUtil
sharding_op_util.cc
sharding_op_util.h
side_effect_util.cc
side_effect_util.h
status.h
status_macros.cc
status_macros.h MakeErrorStream StatusAdaptorForMacros
statusor.h
[+] stream_executor/
types.h
union_find.h UnionFind
util.cc
util.h ConvertedDimensionNumbers DivMod ScopedLoggingTimer SignedIntegerTypeForSize TimerStats UnsignedIntegerTypeForSize
window_util.cc
window_util.h
xla.grpc.pb.cc
xla.grpc.pb.h
xla.pb.cc CompileRequestDefaultTypeInternal CompileResponseDefaultTypeInternal ComputationGraphStatsRequestDefaultTypeInternal ComputationStatsResponseDefaultTypeInternal ComputeConstantGraphRequestDefaultTypeInternal ComputeConstantResponseDefaultTypeInternal CreateChannelHandleRequestDefaultTypeInternal CreateChannelHandleResponseDefaultTypeInternal DebugOptionsDefaultTypeInternal DebugOptions_XlaBackendExtraOptionsEntry_DoNotUseDefaultTypeInternal DeconstructTupleRequestDefaultTypeInternal DeconstructTupleResponseDefaultTypeInternal ExecuteGraphParallelRequestDefaultTypeInternal ExecuteGraphRequestDefaultTypeInternal ExecuteParallelResponseDefaultTypeInternal ExecuteRequestDefaultTypeInternal ExecuteResponseDefaultTypeInternal ExecutionOptionsDefaultTypeInternal GetDeviceHandlesRequestDefaultTypeInternal GetDeviceHandlesResponseDefaultTypeInternal GetShapeRequestDefaultTypeInternal GetShapeResponseDefaultTypeInternal LoadDataRequestDefaultTypeInternal LoadDataResponseDefaultTypeInternal ResetDeviceRequestDefaultTypeInternal ResetDeviceResponseDefaultTypeInternal TransferFromOutfeedRequestDefaultTypeInternal TransferFromOutfeedResponseDefaultTypeInternal TransferToClientRequestDefaultTypeInternal TransferToClientResponseDefaultTypeInternal TransferToInfeedRequestDefaultTypeInternal TransferToInfeedResponseDefaultTypeInternal TransferToServerRequestDefaultTypeInternal TransferToServerResponseDefaultTypeInternal UnpackRequestDefaultTypeInternal UnpackResponseDefaultTypeInternal UnregisterRequestDefaultTypeInternal UnregisterResponseDefaultTypeInternal WaitForExecutionRequestDefaultTypeInternal WaitForExecutionResponseDefaultTypeInternal
xla.pb.h TableStruct_tensorflow_2fcompiler_2fxla_2fxla_2eproto is_proto_enum CompileRequest CompileResponse ComputationGraphStatsRequest ComputationStatsResponse ComputeConstantGraphRequest ComputeConstantResponse CreateChannelHandleRequest CreateChannelHandleResponse DebugOptions DebugOptions_ShapeChecks DebugOptions_StepMarkerLocation DebugOptions_XlaBackendExtraOptionsEntry_DoNotUse DeconstructTupleRequest DeconstructTupleResponse ExecuteGraphParallelRequest ExecuteGraphRequest ExecuteParallelResponse ExecuteRequest ExecuteResponse ExecutionOptions GetDeviceHandlesRequest GetDeviceHandlesResponse GetShapeRequest GetShapeResponse LoadDataRequest LoadDataResponse ResetDeviceRequest ResetDeviceResponse TransferFromOutfeedRequest TransferFromOutfeedResponse TransferToClientRequest TransferToClientResponse TransferToInfeedRequest TransferToInfeedResponse TransferToServerRequest TransferToServerResponse UnpackRequest UnpackResponse UnregisterRequest UnregisterResponse WaitForExecutionRequest WaitForExecutionResponse
xla_data.pb.cc ChannelHandleDefaultTypeInternal CholeskyOptionsDefaultTypeInternal ComputationStatsDefaultTypeInternal ConvolutionDimensionNumbersDefaultTypeInternal CustomCallOutputOperandAliasingDefaultTypeInternal DeviceAssignmentProtoDefaultTypeInternal DeviceAssignmentProto_ComputationDeviceDefaultTypeInternal DeviceHandleDefaultTypeInternal DotDimensionNumbersDefaultTypeInternal ExecutionHandleDefaultTypeInternal ExecutionProfileDefaultTypeInternal FrontendAttributesDefaultTypeInternal FrontendAttributes_MapEntry_DoNotUseDefaultTypeInternal GatherDimensionNumbersDefaultTypeInternal GlobalDataHandleDefaultTypeInternal LayoutProtoDefaultTypeInternal LiteralProtoDefaultTypeInternal OpMetadataDefaultTypeInternal OpMetadata_ProfileInfoDefaultTypeInternal OpShardingDefaultTypeInternal PaddingConfigDefaultTypeInternal PaddingConfig_PaddingConfigDimensionDefaultTypeInternal ParameterReplicationDefaultTypeInternal PrecisionConfigDefaultTypeInternal ProgramShapeProtoDefaultTypeInternal ReplicaGroupDefaultTypeInternal ScatterDimensionNumbersDefaultTypeInternal ShapeProtoDefaultTypeInternal SourceTargetDefaultTypeInternal TileProtoDefaultTypeInternal TriangularSolveOptionsDefaultTypeInternal WhileLoopBackendConfigDefaultTypeInternal WhileLoopBackendConfig_KnownTripCountDefaultTypeInternal WindowDefaultTypeInternal WindowDimensionDefaultTypeInternal
xla_data.pb.h TableStruct_tensorflow_2fcompiler_2fxla_2fxla_5fdata_2eproto is_proto_enum ChannelHandle ChannelHandle_ChannelType CholeskyOptions CompilationEvent ComputationStats ConvolutionDimensionNumbers CustomCallOutputOperandAliasing DeviceAssignmentProto DeviceAssignmentProto_ComputationDevice DeviceHandle DimLevelType DotDimensionNumbers ExecutionHandle ExecutionProfile FftType FrontendAttributes FrontendAttributes_MapEntry_DoNotUse GatherDimensionNumbers GlobalDataHandle LayoutProto LiteralProto OpMetadata OpMetadata_ProfileInfo OpSharding OpSharding_Type PaddingConfig PaddingConfig_PaddingConfigDimension PaddingType ParameterReplication PrecisionConfig PrecisionConfig_Precision PrimitiveType ProfileSource ProfileType ProgramShapeProto RandomAlgorithm RandomDistribution ReplicaGroup ScatterDimensionNumbers ShapeProto SourceTarget TileProto TriangularSolveOptions TriangularSolveOptions_Transpose WhileLoopBackendConfig WhileLoopBackendConfig_KnownTripCount Window WindowDimension

Browse the source code of tensorflow/tensorflow/tensorflow/compiler/xla/ online