Searched defs:program_shape (Results 26 – 39 of 39) sorted by relevance
12
77 const XlaComputation& computation, const ProgramShape& program_shape) { in GetShardedProgramShapes()
127 Status program_shape(xla::ProgramShapeProto* program_shape) override { in program_shape() function in tpu_driver::__anonb00856100111::GrpcCompiledProgramHandle
74 required xla.ProgramShapeProto program_shape = 1; field
41 xla.ProgramShapeProto program_shape = 4; field
328 StaticData* static_data, const xla::ProgramShapeProto* program_shape) { in set_static_data_program_shape()
1117 xla::ProgramShapeProto program_shape; in TEST() local1503 xla::ProgramShapeProto program_shape; in TEST() local1551 xla::ProgramShape program_shape(program_shape_proto); in TEST() local2040 xla::ProgramShapeProto program_shape; in TEST() local
748 std::unique_ptr<xla::ProgramShapeProto> program_shape; in GenerateMetadata() local
247 ProgramShape program_shape; in MakeProgramShape() local707 const ProgramShape& program_shape) { in HumanString()
366 ProgramShape program_shape; in BuildXlaCompilerSubmodule() local
355 xla.ProgramShapeProto program_shape = 4; field
879 ProgramShape program_shape; in ComputeProgramShape() local
3010 ProgramShape program_shape; in to_apply() local
318 ProgramShape program_shape; in GetProgramShape() local586 HloModuleProto* module, const ProgramShape& program_shape, in PopulateInputOutputAlias()3666 ProgramShapeProto* program_shape = entry.mutable_program_shape(); in BuildConstantSubGraph() local
4508 auto program_shape = module->entry_computation()->ComputeProgramShape(); in Run() local