Searched refs:program_shape_ (Results 1 – 4 of 4) sorted by relevance
93 const xla::ProgramShapeProto* program_shape_ = nullptr; variable258 const xla::ProgramShapeProto* ProgramShape() const { return program_shape_; } in ProgramShape()329 static_data->program_shape_ = program_shape; in set_static_data_program_shape()383 const xla::ProgramShapeProto* program_shape_ = nullptr; variable
36 program_shape_(static_data.program_shape_), in XlaCompiledCpuFunction()
145 jit->program_shape_ = in Compile()182 &jit->static_data_, jit->program_shape_.get()); in Compile()
89 std::unique_ptr<const xla::ProgramShapeProto> program_shape_; variable